怎么将dataframe中year列取值为2013的所有行在A和B列的数据提取出来
时间: 2024-11-20 13:38:41 浏览: 16
在Pandas DataFrame中,如果你想从year列值为2013的所有行中提取A列和B列的数据,你可以使用`.loc`、`.iloc`或者布尔索引来实现。这里给出几种方法:
1. 使用布尔索引:
```python
df_2013 = df[df['year'] == 2013] # 筛选出2013年的数据
data_2013 = df_2013[['A', 'B']] # 提取'A'和'B'两列
```
2. 使用`.loc`方法:
```python
data_2013 = df.loc[df['year'] == 2013, ['A', 'B']]
```
这里`['A', 'B']`表示选择A和B这两列。
3. 使用`.iloc`方法(适用于整数索引):
```python
data_2013 = df.iloc[df.index[df['year'] == 2013], [df.columns.get_loc('A'), df.columns.get_loc('B')]]
```
注意这里的`get_loc`函数是为了找到'A'和'B'列的实际整数位置。
以上代码都会返回一个新的DataFrame,其中包含所有2013年对应的A和B列数据。
阅读全文