DataFrame 如何同时取出两列的值
时间: 2024-04-11 20:27:09 浏览: 132
Pandas DataFrame Notes
要同时取出 DataFrame 中的两列值,可以使用以下方法:
1. 使用方括号 [] 运算符,传入列名作为列表来选择两列。例如,假设 DataFrame 名称为 df,想要选择列名为 "column1" 和 "column2" 的两列,可以使用以下代码:
```python
df[['column1', 'column2']]
```
2. 使用 loc 或 iloc 属性来选择两列。loc 属性用于基于标签进行选择,iloc 属性用于基于位置进行选择。以下是两种方法的示例代码:
使用 loc 属性:
```python
df.loc[:, ['column1', 'column2']]
```
使用 iloc 属性:
```python
df.iloc[:, [0, 1]] # 假设 "column1" 是第一列,"column2" 是第二列
```
这些方法将返回一个包含两列的新 DataFrame。如果你只是想要取出两列的值,并不需要返回 DataFrame,可以将这些方法的结果转换为 NumPy 数组或列表。例如,使用 values 属性将结果转换为 NumPy 数组:
```python
df[['column1', 'column2']].values
```
希望这个回答能够帮到你!如果有任何进一步的问题,请随时提问。
阅读全文