从字典元素创建Pandas数据框架

Question

我将尝试从一个字典创建一个pandas数据框。该字典的设置如下：

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

我希望数据框只包含"y1"和"y2"。目前我可以通过以下方式实现：

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

我想知道是否有可能在不使用df.drop()的情况下完成这个任务。

- Lukasz

Answer 1

你可以在 DataFrame 构造函数中指定 columns:

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

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

- John Zwinck

Answer 2

您可以使用一个字典推导式来过滤掉不需要的键，如下所示：

您可以使用字典推导式来过滤掉不需要的键，例如：

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

当然，有很多方法可以从一个字典中删除一个或多个键。

点击这里查看更多信息。

- Stefan