Iterating and validating through a pandas dataframe

Question

I have a pandas DataFrame called 'heat' consisting of 2 columns, A and B

Column A cells hold string values of : 'High' 'Low' and 'Central'.

What I want to achieve is iterate through the dataframe and do the following checks:

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'

Any ideas please?

Answer 1

尝试使用map ：

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