pd.read_csv选列
时间: 2024-02-09 22:04:59 浏览: 25
以下是两种使用pd.read_csv选取列的方法:
1. 使用列名取列:
```python
import pandas as pd
data = pd.read_csv("1.csv")["9"] # 取列名是“9”的那一列
data1 = pd.read_csv("1.csv")[["9","2"]] # 索引以列表的形式出现
```
2. 使用usecols参数取列:
```python
import pandas as pd
data = pd.read_csv("1.csv", usecols=["9"]) # 取列名是“9”的那一列
data1 = pd.read_csv("1.csv", usecols=["9","2"]) # 取列名是“9”和“2”的两列
```
相关问题
pd.read_csv取出指定列
可以使用pd.read_csv()函数的usecols参数来取出指定列。usecols参数可以接受一个列表,列表中包含需要取出的列名或列号。例如,假设我们有一个名为data.csv的文件,其中包含三列数据,分别为A、B、C,我们想要取出B列和C列的数据,可以使用以下代码:
```python
import pandas as pd
# 读取文件,只取出B列和C列的数据
df = pd.read_csv('data.csv', usecols=['B', 'C'])
# 打印取出的数据
print(df)
```
如果我们想要取出第2列和第3列的数据,可以使用以下代码:
```python
import pandas as pd
# 读取文件,只取出第2列和第3列的数据
df = pd.read_csv('data.csv', usecols=[1, 2])
# 打印取出的数据
print(df)
```
pd.read_csv选择读入列数
在使用`pd.read_csv`函数读取CSV文件时,可以通过指定`usecols`参数来选择读入的列数。`usecols`参数可以接受一个列表或者一个函数作为输入。
如果传入一个列表,列表中的元素可以是列名或者列索引,表示只读取指定的列。例如,如果你只想读取CSV文件中的"列1"和"列3",可以这样设置`usecols`参数:
```python
import pandas as pd
data = pd.read_csv('data.csv', usecols=['列1', '列3'])
```
如果传入一个函数,函数会被应用到CSV文件的每一列上,返回值为True的列会被读取。例如,如果你只想读取列名以"列"开头的列,可以这样设置`usecols`参数:
```python
import pandas as pd
data = pd.read_csv('data.csv', usecols=lambda column: column.startswith('列'))
```
这样就只会读取以"列"开头的列。