通过考虑两列，想知道如何在python数据框架中使用startswith吗

Question

我的输入是（A，B有数据，需要在C列中输出）

逻辑是正确的。如果我在A和B列中的前四个字母是INKA或IDKA，那么Out是KAR

同样的方法是INAP或IDAP然后AP和INRJ或IDRJ然后RAJ

输入为A和B，预期输出为C

^{tb1}$

Answer 1

您可以使用np.select：

conditions = [(df["A"].str[:4].isin(["INKA", "IDKA"]))|(df["B"].str[:4].isin(["INKA", "IDKA"])),
              (df["A"].str[:4].isin(["INAP", "IDAP"]))|(df["B"].str[:4].isin(["INAP", "IDAP"])),
              (df["A"].str[:4].isin(["INRJ", "IDRJ"]))|(df["B"].str[:4].isin(["INRJ", "IDRJ"]))]

df["C"] = np.select(conditions, ["KAR", "AP", "RAJ"], None)

或者，您可以使用map和combine_first：

mapper = {"INKA": "KAR", "IDKA": "KAR", "INAP": "AP", "IDAP": "AP", "INRJ": "RAJ", "IDRJ": "RAJ"
df["C"] = df["A"].str[:4].map(mapper).combine_first(df["B"].str[:4].map(mapper))