python array写入到 csv 指定列

在Python中，你可以使用pandas库方便地将数组数据写入CSV文件并指定特定列。Pandas提供了一个叫做`to_csv()`的功能，它可以处理这样的操作。假设你有一个名为`data_array`的数组，你想将其作为CSV文件的某列，首先你需要创建一个包含这个数组的DataFrame。

import pandas as pd

# 假设你的数组 data_array 长度等于另一列 df['existing_column'] 的长度
data_array = ...  # 你的数组内容
df = ...  # 你的DataFrame对象，其中有一个已存在的列（例如 'existing_column'）

# 将数据添加为新列，这里以'new_column'为例
df['new_column'] = data_array

# 现在可以将DataFrame写入CSV文件，并指明你想保留的列
columns_to_write = ['existing_column', 'new_column']
df[columns_to_write].to_csv('output.csv', index=False)