python - 按索引号向列 pandas 数据框添加值

Question

假设我有一个表示数据帧索引的列表，如下所示:

indexlist = [0,1,4]

以及以下数据框:

    Name    Country
0   John    BR
1   Peter   BR
2   Paul    BR
3   James   CZ
4   Jonatan CZ
5   Maria   DK

我需要在此数据框上创建一个名为“Is it valid?”的列如果索引行在列表中，那将添加"is"。否则将添加“否”，导致此数据框:

    Name    Country  Is it valid?
0   John    BR       Yes
1   Peter   BR       Yes
2   Paul    BR       No
3   James   CZ       No
4   Jonatan CZ       Yes
5   Maria   DK       No

有什么办法可以做到吗？

谢谢!

Answer 1

您可以使用 isin 作为索引，您通常会在 Series 中使用它，它实际上创建了一个真值数组，您可以将其传递给 np.where使用 true 和 false 值，将结果分配为一列。

df['Is it valid?'] = np.where(df.index.isin(indexlist), 'Yes', 'No')

输出:

      Name Country Is it valid?
0     John      BR          Yes
1    Peter      BR          Yes
2     Paul      BR           No
3    James      CZ           No
4  Jonatan      CZ          Yes
5    Maria      DK           No