要按照给定列表的顺序选择pandas DataFrame的行并保留原始索引,可以使用DataFrame的loc
属性和提供的列表作为索引。下面是一个示例代码:
import pandas as pd
# 创建示例数据
data = {'Name': ['Tom', 'Nick', 'John', 'Alex'],
'Age': [25, 30, 35, 40],
'Country': ['USA', 'Canada', 'UK', 'Australia']}
df = pd.DataFrame(data)
# 给定的列表顺序
order = [3, 1, 2, 0]
# 按照给定列表的顺序选择行,并保留原始索引
df_selected = df.loc[order]
print(df_selected)
输出结果:
Name Age Country
3 Alex 40 Australia
1 Nick 30 Canada
2 John 35 UK
0 Tom 25 USA
在上述代码中,我们首先创建了一个示例DataFrame df
。然后,我们定义了一个给定列表顺序 order
,其中包含了我们想要选择行的索引顺序。最后,我们使用df.loc[order]
选择了按照给定列表顺序的行,并将结果赋值给变量df_selected
。
上一篇:按照给定键将对象分组在数组中
下一篇:按照给定列表对数据集进行排序。