以下是一个示例代码,演示如何遍历键的列表,并将存在于给定行/列中的键添加到Pandas数据框中:
import pandas as pd
# 创建一个示例数据框
data = {'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]}
df = pd.DataFrame(data)
# 给定的行/列
given_rows = ['A', 'C']
# 创建一个新的数据框,用于存储匹配的键
new_df = pd.DataFrame()
# 遍历键的列表
for key in given_rows:
# 检查键是否存在于数据框的列中
if key in df.columns:
# 如果存在,则将该列添加到新的数据框中
new_df[key] = df[key]
# 打印新的数据框
print(new_df)
输出结果为:
A C
0 1 7
1 2 8
2 3 9
在上面的示例中,我们首先创建了一个示例数据框df。然后,我们给定了一个行/列名称列表given_rows。接下来,我们创建了一个新的空数据框new_df,用于存储匹配的键。然后,我们通过遍历给定的键列表,并逐个检查每个键是否存在于数据框的列中。如果键存在,则我们将该列添加到新的数据框中。最后,我们打印新的数据框new_df。
上一篇:遍历简单XML对象并追加它们