python - 使用索引设置 Pandas 列的字符串值

Question

我正在尝试在我的数据框中生成一个新列，其中为不同的索引值分配了一个字符串。我不够熟练，无法使用循环来完成，所以我尝试了子集和索引，看看我是否可以让我的生活更轻松。

df['event_type'] = []

A = df[df['flags'].isin(['A'])].index
B= df[df['flags'].isin(['B'])].index
C = df[df['flags'].isin(['B'])].index


df.loc[A, 'event_type'] = 'Condition One'
df.loc[B, 'event_type'] = 'Condition Two'
df.loc[C, 'event_type'] = 'Condition Three'

我已经多次得到值的长度与索引的长度不匹配，并且 Index.name 必须是可散列类型错误好几次了。我只想将这些字符串分配给新列中的这些索引，列中的所有其他值都可以是 Nan。

标签： pythonpandasstringdataframeindexing

Answer 1

你可以使用Series.map，

df['event_type'] = df.flags.fillna('').map(
    {"A": "Condition One", "B": "Condition Two", "C": "Condition Three"}
)