python保存csv指定列的数据 具体例子 详细

当你想要保存CSV文件中的指定列数据时，你可以使用Python的pandas库来实现。以下是一个更详细的例子：

import pandas as pd

# 读取CSV文件
df = pd.read_csv('example.csv')

# 打印原始数据
print("原始数据：")
print(df)

# 选择指定列
selected_columns = ['column1', 'column2']  # 选择要保存的列名
df_selected = df[selected_columns]

# 打印选择的列数据
print("\n选择的列数据：")
print(df_selected)

# 保存为新的CSV文件
df_selected.to_csv('selected_columns.csv', index=False)

# 打印保存的CSV文件内容
print("\n保存的CSV文件内容：")
saved_df = pd.read_csv('selected_columns.csv')
print(saved_df)

在上面的例子中，我们首先使用`pd.read_csv()`函数读取名为`example.csv`的CSV文件，并将数据存储在一个名为`df`的DataFrame对象中。

然后，我们定义一个`selected_columns`列表，其中包含你想要保存的列名。

接下来，我们通过从原始DataFrame中选择指定列创建一个新的DataFrame `df_selected`。

然后，我们打印出原始数据和选择的列数据，以便你可以查看它们。

最后，我们使用`to_csv()`函数将新的DataFrame保存为名为`selected_columns.csv`的CSV文件。`index=False`参数表示不保存行索引。

最后，我们使用`pd.read_csv()`函数再次读取保存的CSV文件，并打印出它的内容，以验证保存是否成功。

你可以根据需要修改列名和文件名。希望对你有所帮助！