如何使用 python 获取列表值和列？

Question

为这个菜鸟问题道歉，但是....

我有一个数据框（在 python 中使用 pandas，带有标题），如下所示：

   Pink     Blue    Yellow   Green  
 --------- ------- -------- ------- 
  Jackson   Bryan   Bruce    Benji  
  James     Jonah   Jenny    Kevin  
  Tyson     Jenna   Mike     none   
  Kendall   Amy     Alison   none   
  Ben       none    none     none   

如何获得如下所示的团队和名称列表？

  Pink    Jackson  
  Pink    James    
  Blue    Bryan    
  ...              
  Green   Kevin    

Answer 1

除了较早的答案，我建议过滤掉 None 的：

for t in filter(lambda x: x[1], pd.melt(df).values.tolist()):
    print('{:8} {}'.format(*t))

Answer 2

如果我猜对了，您希望拥有所有列名 + 值对。 pd.melt 可以为您做到这一点：

 df = pd.DataFrame({"Pink": ["Jackson", "James"], "Blue": ["Bryan", None]})
 pd.melt(df)

结果：

  variable    value
0     Pink  Jackson
1     Pink    James
2     Blue    Bryan
3     Blue     None

如果你只想要一个二维 numpy 数组，你可以这样做：

pd.melt(df).values

结果：

array([['Pink', 'Jackson'],
       ['Pink', 'James'],
       ['Blue', 'Bryan'],
       ['Blue', None]], dtype=object)

正如 matman9 所指出的，只需在 .values 之后添加一个 .tolist() 即可获得 python 列表。