python把数组中的某些元素写入数据帧
时间: 2023-07-25 12:39:30 浏览: 78
向数组中插入元素
如果你想将数组中的某些元素写入数据帧,可以按照以下步骤操作:
1. 导入 pandas 库:
```
import pandas as pd
```
2. 定义需要写入数据帧的数组和需要写入的列名:
```
array = [1, 2, 3, 4, 5]
col_name = 'column_name'
```
其中,`array` 为你要写入数据帧的数组,`col_name` 为你要写入的列名。
3. 将数据帧读入内存:
```
df = pd.read_csv('filename.csv')
```
其中,`filename.csv` 为你要读入的数据帧文件名。
4. 将数组中的元素写入数据帧:
```
df[col_name] = array
```
5. 将数据帧写入文件(可选):
```
df.to_csv('filename.csv', index=False)
```
其中,`filename.csv` 为你要写入的数据帧文件名。
完整示例代码:
```
import pandas as pd
array = [1, 2, 3, 4, 5]
col_name = 'column_name'
df = pd.read_csv('filename.csv')
df[col_name] = array
df.to_csv('filename.csv', index=False)
```
以上代码将数组 `array` 写入数据帧 `df` 中名为 `column_name` 的列,并将结果写入文件 `filename.csv` 中。
阅读全文