如何將數據框的列轉換為具有不同長度的列表？

Question

我有一個帶有不同長度列的數據框。 看起來像這樣：

Croptypes   VariableInputs   FixedInputs
barley      fertilizer       land
rapeseed                     labor
wheat                        capital

當我通過以下代碼從列中列出元素時：

Croptypes = list(df['Croptypes'])
VariableInputs = list(df['VariableInputs'])

我得到這個結果。

['barley', 'rapeseed', 'wheat']
['fertilizer', nan, nan]

如何告訴“列表”不列出空單元格？

我預計：

['barley', 'rapeseed', 'wheat']
['fertilizer']

Answer 1

使用dropna並tolist ：

>>> df['VariableInputs'].dropna().tolist()
['fertilizer']
>>> 

或對於每一列：

print({k: v.dropna().tolist() for k, v in df.to_dict('s').items()})

輸出：

{'Croptypes': ['barley', 'rapeseed', 'wheat'], 'VariableInputs': ['fertilizer'], 'FixedInputs': ['land', 'labor', 'capital']}

Answer 2

您可以為列表的每個列創建字典，其值如下：

d = {col:df[col].dropna().tolist() for col in df}

d
{'Croptypes': ['barley', 'rapeseed', 'wheat'],
 'VariableInputs': ['fertilizer'],
 'FixedInputs': ['land', 'labor', 'capital']}