pandas提取列名为a和b的两列

可以使用pandas的loc方法来提取列名为a和b的两列。假设你的数据集中有列名为'a'、'b'、'c'和'd'，你只想要提取列名为'a'和'b'的两列，可以使用以下代码：

import pandas as pd

# 读取数据集
df = pd.read_csv('your_data.csv')

# 提取列名为'a'和'b'的两列
filtered_df = df.loc[:, ['a', 'b']]

在这个例子中，我们使用了pandas的loc方法，它可以通过行标签和列标签来访问数据。我们传递了一个包含所有行和列名为'a'和'b'的列表作为loc方法的第二个参数，这样就可以提取出这两列。最后，我们将它们存储在`filtered_df`中。