python保存csv指定列的数据 具体例子
时间: 2023-12-17 13:05:31 浏览: 85
使用python获取csv文本的某行或某列数据的实例
5星 · 资源好评率100%
当你想要保存CSV文件中的指定列数据时,你可以使用Python的pandas库来实现。以下是一个具体的例子:
```python
import pandas as pd
# 读取CSV文件
df = pd.read_csv('example.csv')
# 选择指定列
selected_columns = ['column1', 'column2'] # 选择要保存的列名
df_selected = df[selected_columns]
# 保存为新的CSV文件
df_selected.to_csv('selected_columns.csv', index=False)
```
在上面的例子中,我们首先使用`pd.read_csv()`函数读取名为`example.csv`的CSV文件,并将数据存储在一个名为`df`的DataFrame对象中。
然后,我们定义一个`selected_columns`列表,其中包含你想要保存的列名。
接下来,我们通过从原始DataFrame中选择指定列创建一个新的DataFrame `df_selected`。
最后,我们使用`to_csv()`函数将新的DataFrame保存为名为`selected_columns.csv`的CSV文件。`index=False`参数表示不保存行索引。
你可以根据需要修改列名和文件名。希望对你有所帮助!
阅读全文