python取某几列
时间: 2023-11-14 08:12:02 浏览: 534
在Python中,我们可以使用pandas库来选取某几列。具体方法如下:
1. 选取指定列:使用DataFrame的loc方法,指定要选取的列名即可。例如,选取名为column1和column2的两列数据的代码如下:
```
import pandas as pd
data = pd.DataFrame({'column1': [1, 2, 3], 'column2': [4, 5, 6], 'column3': [7, 8, 9]})
selected_data = data.loc[:, ['column1', 'column2']]
```
2. 选取符合条件的列:使用DataFrame的filter方法,指定要筛选的列名的条件即可。例如,选取以column1开头的所有列的代码如下:
```
import pandas as pd
data = pd.DataFrame({'column1_a': [1, 2, 3], 'column1_b': [4, 5, 6], 'column2': [7, 8, 9]})
selected_data = data.filter(regex='^column1')
```
3. 选取某个范围内的列:使用DataFrame的iloc方法,指定要选取的列的范围即可。例如,选取第1列至第3列数据的代码如下:
```
import pandas as pd
data = pd.DataFrame({'column1': [1, 2, 3], 'column2': [4, 5, 6], 'column3': [7, 8, 9]})
selected_data = data.iloc[:, 0:2]
```
阅读全文