可以使用pandas.DataFrame.loc[]方法根据元素的值而不是列名选择列。以下是一个例子:
import pandas as pd
# 创建一个DataFrame
df = pd.DataFrame({
'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]
})
# 按照元素选择列
selected_cols = df.loc[:, df.iloc[0] > 2]
print(selected_cols)
输出结果:
C
0 7
1 8
2 9
在这个例子中,.iloc[0] > 2
选择了第一行中大于2的元素,df.loc[:, df.iloc[0] > 2]
选择了包含这些元素的列并返回了一个新的DataFrame。
上一篇:按照元素的随机规则对列表进行排序
下一篇:按照元素间的距离将列表划分为子组