通過 Pandas 數據框迭代和驗證

Question

我有一個名為“heat”的Pandas DataFrame，由2 列 A 和 B 組成

A 列單元格包含以下字符串值：“高”、“低”和“中央”。

我想要實現的是遍歷數據框並執行以下檢查：

If string at column A ends with 'gh', fill column B with 'Yes'

If string at column A ends with 'ow', fill column B with 'No'

If string at column A ends with 'al', fill column B with 'Maybe'

請問有什么想法嗎？

Answer 1

嘗試使用map ：

heat["B"] = heat["A"].str[-2:].map({"gh": "Yes", "ow": "No", "al": "Maybe"})