Question

假设我们有以下字典，键是元组，值是列表：

dict_temp  = {('first', 'line'): [1, 2], ('second', 'line'): [1, 21, 11]}

我想创建一个包含 2+n 列的数据框：来自元组的 Col1 和 Col2，以及来自列表中每个元素的 Col3、Col4...，它们应该如下所示：

    0       1       2   3   4
0   first   line    1   2   NaN
1   second  line    1   21  11.0

因此，我还需要用列表拆分列。我所做的是：

dict_temp  = {('first', 'line'): [1, 2], ('second', 'line'): [1, 21, 11]}
df_left    = pd.Series(dict_temp).reset_index()
df_right   = pd.DataFrame(df_left[0].tolist())
df_left    = df_left.drop([0], axis=1) # Drop column
df         = pd.concat([df_left, df_right], axis=1)
#df.columns = [0, 1, 2, 3, 4]
df

输出为：

level_0 level_1     0   1   2
0   first   line    1   2   NaN
1   second  line    1   21  11.0

虽然我已经处理了 this question 和 that question 启发的问题，但我认为应该有更好的 Pythonic 方法来处理它。

Answer 1

尝试将这种 list 推导式与列表解包结合使用，然后只需将其转换为 DataFrame：

lst = [[*k] + v for k, v in dict_temp.items()]
print(pd.DataFrame(lst))

输出：

        0     1  2   3     4
0   first  line  1   2   NaN
1  second  line  1  21  11.0

从字典创建数据帧，其中键是元组，值是列表

1 个答案: