要取DataFrame中的多列数据
时间: 2024-06-13 18:05:30 浏览: 87
pandas DataFrame 根据多列的值做判断,生成新的列值实例
5星 · 资源好评率100%
要取DataFrame中的多列数据,可以使用Pandas中的iloc或loc方法。其中,iloc方法可以通过指定行和列的位置来选取数据,loc方法可以通过指定行和列的标签来选取数据。具体操作如下:
1. 选取连续多列数据:df.iloc[:, start_col:end_col],其中start_col和end_col分别为起始列和结束列的位置,选取的数据包括起始列和结束列。
2. 选取不连续多列数据:df.loc[:, ['col1', 'col2', ...]],其中'col1', 'col2', ...为需要选取的列的标签。
下面是一个示例代码:
```
import pandas as pd
# 创建DataFrame
data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
'age': [25, 30, 35, 40],
'gender': ['F', 'M', 'M', 'M'],
'score': [80, 90, 85, 95]}
df = pd.DataFrame(data)
# 选取连续多列数据
df1 = df.iloc[:, 1:3]
print(df1)
# 选取不连续多列数据
df2 = df.loc[:, ['name', 'score']]
print(df2)
```
输出结果如下:
```
age gender
0 25 F
1 30 M
2 35 M
3 40 M
name score
0 Alice 80
1 Bob 90
2 Charlie 85
3 David 95
```
阅读全文