要按行特定列表选择pandas数据帧列,可以使用loc
或iloc
方法。下面是使用这两种方法的代码示例:
使用loc
方法:
import pandas as pd
# 创建一个示例数据帧
data = {'Name': ['Tom', 'Nick', 'John', 'Elon'],
'Age': [28, 32, 45, 39],
'Country': ['USA', 'Canada', 'UK', 'USA']}
df = pd.DataFrame(data)
# 定义要选择的列列表
columns_to_select = ['Name', 'Country']
# 使用loc方法选择特定列表的列
selected_columns = df.loc[:, columns_to_select]
# 打印选择的列
print(selected_columns)
使用iloc
方法:
import pandas as pd
# 创建一个示例数据帧
data = {'Name': ['Tom', 'Nick', 'John', 'Elon'],
'Age': [28, 32, 45, 39],
'Country': ['USA', 'Canada', 'UK', 'USA']}
df = pd.DataFrame(data)
# 定义要选择的列索引列表
column_indices_to_select = [0, 2]
# 使用iloc方法选择特定列表的列
selected_columns = df.iloc[:, column_indices_to_select]
# 打印选择的列
print(selected_columns)
以上两个示例中,我们创建了一个示例数据帧,然后定义了要选择的列列表或列索引列表。使用loc
方法时,我们使用:
选择所有行,然后通过传递列列表来选择特定的列。使用iloc
方法时,我们同样使用:
选择所有行,然后通过传递列索引列表来选择特定的列。最后,我们打印选择的列。
上一篇:按行索引将df1和df2连接
下一篇:按行填充Numpy数组