如何使用 Pandas 进行数据写入?
步骤:
-
创建 DataFrame: 使用
DataFrame()
函数创建 DataFrame 对象。 -
添加数据: 使用
DataFrame.append()
或DataFrame.loc
等方法添加数据。 -
设置索引: 使用
set_index()
方法设置 DataFrame 的索引。 -
设置列名: 使用
columns
参数设置 DataFrame 的列名。 -
写入数据: 使用
to_csv()
或to_excel()
方法将 DataFrame 写入 CSV 或 Excel 文件中。
示例:
import pandas as pd
# 创建 DataFrame
data = {'name': ['John', 'Mary', 'Bob'], 'age': [25, 30, 35]}
df = pd.DataFrame(data)
# 添加数据
df.loc['Jane', 'age'] = 28
# 设置索引
df.set_index('name', inplace=True)
# 设置列名
df.columns = ['age', 'name']
# 写入数据
df.to_csv('data.csv', index=False)
结果:
name age
John 25
Mary 30
Bob 35
Jane 28
其他方法:
-
DataFrame.insert()
:添加数据到指定位置。 -
DataFrame.drop()
:删除数据。 -
DataFrame.groupby()
:对数据进行分组。 -
DataFrame.agg()
:对数据进行聚合。