使用字符串获取某些列 pandas - Python

在pandas中，我们可以使用字符串获取特定的列。这种方法既简单又灵活，特别是在处理一些大型的数据集时。

使用列名获取列

要使用列名获取列，我们可以使用DataFrame对象的列名属性。例如：

import pandas as pd

df = pd.read_csv('sales_data.csv')

#获取Sales列
sales = df['Sales']

print(sales)

通过这种方法，我们可以轻松获取一个或多个列。

使用列索引获取列

如果我们不知道某个列的名称，但知道它的索引位置，使用iloc属性可以获取该列。例如：

import pandas as pd

df = pd.read_csv('sales_data.csv')

#获取第1列
col1 = df.iloc[:, 0]

print(col1)

在这个例子中，我们使用iloc第一个参数表示获取所有行，第二个参数为0表示获取第一列。

使用字符串获取一组列

如果我们只知道列名的一部分，或者想要获取一组列，我们可以使用filter()函数。例如：

import pandas as pd

df = pd.read_csv('sales_data.csv')

#获取所有以'Sales'开头的列
sales_cols = df.filter(like='Sales')

print(sales_cols)

在这个例子中，我们用了filter()函数并且参数like='Sales'来获取所有以'Sales'开头的列。

通过这些例子，我们可以看到在pandas中使用字符串获取特定的列是多么方便和灵活。