Pandas列选择的Pandas优化

Question

我用的是Python，我有一个熊猫DataFrame。此DataFrame (产品)是从CSV文件创建的。CSV文件可能包含country列，也可能不包含。如果它存在，我需要使用该值，否则我将使用默认的国家/地区值。

我正在尝试通过一个熊猫操作来减少这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')
    ...