帮我写一段python代码,有一个数组和一个csv文件,数组对应一个column,要利用数组作为索引,挑选出csv中的数据
时间: 2023-05-18 10:07:19 浏览: 72
用python给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 列,并将结果打印出来。
阅读全文