python保存csv指定列的数据 具体例子
时间: 2023-12-17 15:05:31 浏览: 86
当你想要保存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`参数表示不保存行索引。
你可以根据需要修改列名和文件名。希望对你有所帮助!
相关问题
python保存csv指定列的数据 具体例子 详细
当你想要保存CSV文件中的指定列数据时,你可以使用Python的pandas库来实现。以下是一个更详细的例子:
```python
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文件,并打印出它的内容,以验证保存是否成功。
你可以根据需要修改列名和文件名。希望对你有所帮助!
python csv在指定列写入
如果你想在 CSV 文件的指定列中写入数据,你可以使用 Python 自带的 csv 模块。以下是一个例子,假设我们有一个名为 "data.csv" 的 CSV 文件,其中包含三列数据 "姓名"、"年龄" 和 "性别":
```
姓名,年龄,性别
张三,18,男
李四,20,女
王五,22,男
```
现在,我们想在 "年龄" 列中写入新的数据。可以使用以下代码:
```python
import csv
# 打开 CSV 文件
with open('data.csv', 'r', newline='') as file:
reader = csv.reader(file)
# 读取 CSV 文件中的数据
rows = []
for row in reader:
rows.append(row)
# 在指定列中写入数据
for i in range(1, len(rows)):
rows[i][1] = str(int(rows[i][1]) + 1)
# 保存修改后的数据到 CSV 文件中
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(rows)
```
以上代码将会把 "年龄" 列中的数据都加 1,并写回到原始的 CSV 文件中。
阅读全文