Pandas列选择的Pandas优化

Question

我使用Python并拥有Pandas DataFrame。此DataFrame（产品）是从CSV文件创建的。 CSV文件可能包含或不包含country列。如果它存在，我需要使用该值，否则我将使用默认的国家/地区值。

我试图用一个pandas操作减少这3行，有什么建议吗？

CONTEXT_COUNTRY = 'US'    
for _, product in products.iterrows():
    item_query = _GetItemQuery(product[DESCRIPTION])
    # Country of the Product. Help return better results.
    country = CONTEXT_COUNTRY
    if COUNTRY in products.columns:
      country = product[COUNTRY]

Answer 1

类似于dictionary.get，Series也有get方法。

...
for _, product in products.iterrows():
    ...
    country = products.get(COUNTRY, 'default')
    ...