帮我写一段python代码,有一个数组和一个csv文件,数组对应一个column,要利用数组作为索引,挑选出csv中的数据
时间: 2023-05-18 18:07:19 浏览: 76
python3使用pandas1.5.3给csv文件添加一列数据
当然可以,以下是一段示例代码:
```python
import pandas as pd
# 假设数组为 [1, 3, 5, 7]
arr = [1, 3, 5, 7]
# 读取 csv 文件
df = pd.read_csv('data.csv')
# 选取指定列,并使用数组作为索引
result = df.iloc[:, arr]
# 打印结果
print(result)
```
其中,`data.csv` 是你要读取的 csv 文件名,`arr` 是你要使用的数组。这段代码会选取 csv 文件中的第 1、3、5、7 列,并将结果打印出来。
阅读全文