有一个列表[001,002,003,004],要在CSV文件的二维数组中,查找包含该列表元素对应的列值
时间: 2023-05-20 15:01:09 浏览: 107
获取二维数组的一行或一列.xls
5星 · 资源好评率100%
可以使用Python的pandas库来实现这个功能。首先,读取CSV文件并将其转换为pandas的DataFrame对象:
```python
import pandas as pd
df = pd.read_csv('filename.csv')
```
然后,使用isin()方法查找包含列表元素对应的列值:
```python
cols = ['001', '002', '003', '004']
result = df.loc[:, df.columns.isin(cols)]
```
这将返回一个包含所有包含列表元素对应的列值的DataFrame对象。
阅读全文