python筛选指定列
时间: 2023-10-25 16:22:04 浏览: 79
python 提取文件指定列的方法示例
在 Python 中,可以使用 Pandas 库来对数据进行筛选和处理。如果你有一个 DataFrame 对象,可以使用该对象的 loc 或 iloc 方法来选择指定列。
loc 方法是基于标签的索引,而 iloc 方法是基于位置的索引。
下面是一个例子,假设你有一个包含三列数据的 DataFrame 对象,你想要选择第一列和第三列,可以使用 loc 方法:
```python
import pandas as pd
# 创建一个 DataFrame 对象
data = {'col1': [1, 2, 3, 4], 'col2': [5, 6, 7, 8], 'col3': [9, 10, 11, 12]}
df = pd.DataFrame(data)
# 使用 loc 方法选择指定列
df_new = df.loc[:, ['col1', 'col3']]
print(df_new)
```
输出:
```
col1 col3
0 1 9
1 2 10
2 3 11
3 4 12
```
这里的 `:` 表示选择所有行,`['col1', 'col3']` 表示选择列名为 `col1` 和 `col3` 的两列。
阅读全文