博客

在 Python 中创建DataFrame的方法

  • Post category:Python

在Python中，有多种方法可以用来创建DataFrame。常用的方法包括：

  1. 从列表或numpy数组中创建DataFrame：

可以通过传递列表或numpy数组来创建一个DataFrame，其中列表或数组的长度应该相同。例如：

import pandas as pd
import numpy as np

data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
        'age': [25, 30, 35, 40],
        'salary': [5000, 6000, 7000, 8000]}

df = pd.DataFrame(data)
print(df)

输出：

       name  age  salary
0     Alice   25    5000
1       Bob   30    6000
2   Charlie   35    7000
3     David   40    8000

也可以通过numpy数组创建DataFrame：

data = np.array([['Alice', 25, 5000],
                 ['Bob', 30, 6000],
                 ['Charlie', 35, 7000],
                 ['David', 40, 8000]])

df = pd.DataFrame(data, columns=['name', 'age', 'salary'])
print(df)

输出：

       name age salary
0     Alice  25   5000
1       Bob  30   6000
2   Charlie  35   7000
3     David  40   8000
  1. 从字典中创建DataFrame：

可以通过字典来创建DataFrame，其中字典的键将成为列名，字典的值将成为行的值。例如：

data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
        'age': [25, 30, 35, 40],
        'salary': [5000, 6000, 7000, 8000]}

df = pd.DataFrame.from_dict(data)
print(df)

输出：

   name  age  salary
0  Alice   25    5000
1    Bob   30    6000
2    Charlie   35    7000
3  David   40    8000
  1. 从文件中读取数据创建DataFrame：

可以从文件中读取数据创建DataFrame，常见的文件格式包括csv、Excel等。例如：

df = pd.read_csv('data.csv')
print(df)

输出：

   id    name  age  salary
0   1   Alice   25    5000
1   2     Bob   30    6000
2   3    Kate   35    7000
3   4   David   40    8000

以上便是在Python中创建DataFrame的几种常见方法，开发者可以根据需求选择不同的方法来创建DataFrame。

Tags: ,