python - Pandas :选择列，如果不存在则默认

Question

假设我有以下 DataFrame:

>>> df
     val1 val2 val3
key
  1     1    1    1
  2     2    2    2
  3     3    3    3

现在我想选择列 val1、val2 和(这里是关键:)val4

>>> df[["val1", "val2", "val4"]]
KeyError: "['val4'] not in index"

我想要什么:

>>> df.something(something)
     val1 val2 val4
key
  1     1    1  NaN
  2     2    2  NaN
  3     3    3  NaN

Answer 1

IIUC 重新索引

df.reindex(columns=["val1", "val2", "val4"])
Out[431]: 
     val1  val2  val4
key                  
1       1     1   NaN
2       2     2   NaN
3       3     3   NaN

.loc 也可以做到这一点，但会发出警告:将喜欢的列表传递给 .loc 或 [] 任何缺少的标签将在将来引发 KeyError，您可以使用 .reindex( ) 作为替代方案。

df.loc[:,["val1", "val2", "val4"]]