如何使用Pandas在python中使用if语句创建表?

Question

我需要帮助在我的代码。我的想法是在输入中取一个值(示例:1)，我的代码应该返回分别关于我在输入中选择的表，以便稍后使用此值创建一个超级表(使用pd.merge())，根据输入中的值包含我的所有值。

这是我真实代码的一个例子:

c = int(input('num'))
if c == 1:
aa = [1,2]
aaa = pd.DataFrame(aa,columns=['A'])

else:
aaa = False
if c == 2:
bb = [3,4]
bbb = pd.DataFrame(bb,columns=['B'])

else:
bbb = False
if c == 1:
dd = [5,6]
ddd = pd.DataFrame(dd,columns=['D'])

else:
ddd =False
p = [aaa,bbb,ddd]
for k in p:
if k != False:
print(k)

Answer 1

有几个选项。如果单个数据帧很小，可以将它们放在字典中，然后检索相关的数据帧。

import pandas as pd
d = {
k: pd.DataFrame(range(k, k+1), columns=list("abcde")[k:k+1])
for k in range(4)
}
c = int(input('num'))
print(d.get(c))