向数据帧添加具有单个分类值的列

Question

我有一个pandas.DataFrame{}，想添加一个新列col，其中有一个值"hello"。我希望本专栏的数据类型为category，单一类别为"hello"。我可以做到以下几点

df["col"] = "hello"
df["col"] = df["col"].astype("catgegory")

1. 我真的需要写df["col"]三次才能达到这个目的吗
2. 在第一行之后，我担心中间数据帧df在新列转换为category之前可能会占用大量空间。（数据帧相当大，有数百万行，值"hello"实际上是一个更长的字符串。）

在避免上述问题的同时，是否还有其他直接的、简短的、快速的方法来实现这一点

另一种解决办法是

df["col"] = pd.Categorical(itertools.repeat("hello", len(df)))

但是它需要itertools和len(df)的使用，我不确定内存使用情况如何

Answer 1

这个解决方案肯定能解决第一点，但不一定能解决第二点：

df['col'] = pd.Categorical(('hello' for i in len(df)))

本质上

• 我们首先创建一个长度等于df中记录数的“hello”生成器
• 然后我们将其传递给pd.Categorical，使其成为一个分类列