python - 如何将元组值设置为 Pandas 数据框？

Question

我想做的应该很简单。本质上，我有一些数据框，我需要为某些列分配一些元组值。

例如:

pd_tmp = pd.DataFrame(np.random.rand(3,3))
pd_tmp["new_column"] = ("a",2)

我只需要一个带有元组值的新列，我该怎么办？

ValueError: Length of values does not match length of index

前面的代码出现错误。

Answer 1

您可以将元组包装在列表中:

import pandas as pd
pd_tmp = pd.DataFrame(np.random.rand(3,3))
pd_tmp["new_column"] = [("a",2)] * len(pd_tmp)

pd_tmp
#          0           1           2    new_column
#0  0.835350    0.338516    0.914184    (a, 2)
#1  0.007327    0.418952    0.741958    (a, 2)
#2  0.758607    0.464525    0.400847    (a, 2)