dataframe取特定几列有几种方式
时间: 2024-10-09 16:12:48 浏览: 73
pandas按行按列遍历Dataframe的几种方式
DataFrame是一个二维表格数据结构,在Python的数据分析库pandas中非常常见。从DataFrame中选取特定几列有多种方式:
1. **列名直接选择**:你可以通过列名列表直接指定列名来获取。例如:
```python
df = pd.DataFrame({'A': [1, 2], 'B': [3, 4], 'C': [5, 6]})
cols_to_select = ['A', 'C']
selected_cols = df[cols_to_select]
```
2. **iloc索引**:可以使用整数位置来选择列,`loc`用于标签索引,而`iloc`则支持基于位置的选择。例如:
```python
selected_cols_iloc = df.iloc[:, [0, 2]] # 选取第0和2列
```
3. **布尔索引**:利用条件筛选出想要的列,比如:
```python
bool_mask = df.columns.isin(['A', 'C']) # 创建一个布尔数组
selected_cols_bool = df.loc[:, bool_mask] # 取满足条件的所有列
```
4. **使用`.copy()`方法**:如果你想保留原DataFrame不变,并创建一个新的只包含某些列的DataFrame,可以这样做:
```python
df_selected = df[['A', 'C']].copy()
```
阅读全文