如何在 Pandas 中檢查一個列是否存在?


為了檢查 Pandas DataFrame 中是否存在一個列,我們可以按照以下步驟操作 −

步驟

  • 建立一個二維、大小可變、潛在的異構表格資料,**df**。

  • 列印輸入 DataFrame,**df**。

  • 使用列名初始化一個**col**變數。

  • 建立一個使用者自定義函式**check()**來檢查 DataFrame 中是否存在一個列。

  • 使用有效的列名呼叫**check()**方法。

  • 使用無效的列名呼叫**check()**方法。

示例

 線上演示

import pandas as pd

def check(col):
   if col in df:
      print "Column", col, "exists in the DataFrame."
   else:
      print "Column", col, "does not exist in the DataFrame."

df = pd.DataFrame(
   {
      "x": [5, 2, 1, 9],
      "y": [4, 1, 5, 10],
      "z": [4, 1, 5, 0]
   }
)

print "Input DataFrame is:
", df col = "x" check(col) col = "a" check(col)

輸出

Input DataFrame is:
   x  y  z
0  5  4  4
1  2  1  1
2  1  5  5
3  9 10  0
Column x exists in the DataFrame.
Column a does not exist in the DataFrame.

更新於: 2021 年 8 月 30 日

9000+ 次瀏覽

開啟您的 職業生涯

完成課程即可獲得認證

開始
廣告