从字典元素创建Pandas数据框

Question

我试图从字典中创建一个pandas数据帧。字典设置为

nvalues = {"y1": [1, 2, 3, 4], "y2": [5, 6, 7, 8], "y3": [a, b, c, d]}

我希望数据框只包含"y1"和＆＃34; y2"。到目前为止，我可以使用

完成此任务

df = pd.DataFrame.from_dict(nvalues)
df.drop("y3", axis=1, inplace=True)

我想知道是否有可能在没有df.drop()

的情况下完成此任务

Answer 1

您可以在columns构造函数中指定DataFrame：

pd.DataFrame(nvalues, columns=('y1', 'y2'))

   y1  y2
0   1   5
1   2   6
2   3   7
3   4   8

Answer 2

您可以使用dictionary comprehension过滤掉不需要的keys，如下所示：

df = pd.DataFrame.from_dict({k: v for k, v in nvalues.items() if k != 'y3'})

得到：

   y1  y2
0   1   5
1   2   6
2   3   7
3   4   8

当然，dictionary中有remove one or more keys种方法。