如何替换数据框中的特定元素

时间:2020-04-27 09:44:18

标签: python pandas

我有一个数据框,其中一列的国家/地区名称前有数字,但不是所有条目。我要替换它们。以下是代码的一部分

import pandas as pd
energy = pd.read_excel('Energy Indicators.xls')
energy = energy.take(list(range(18,243)))
energy = energy.drop(labels = ['Unnamed: 0','Unnamed: 1'],axis=1)
energy.columns = ['Country','Energy Supply','Energy Supply per Capita','% Renewable']
energy['Energy Supply'] = energy['Energy Supply'].replace('...',np.nan)
energy['Energy Supply per Capita'] = energy['Energy Supply per 
                                            Capita'].replace('...',np.nan)
energy['Energy Supply'] = energy['Energy Supply']*1000000
def get_energy(item):
    if ' (' in item:
       return item[:item.find(' (')]
    elif '[' in item:
       return item[:item.find('[')]
    else:
       return item
energy['Country'] = energy['Country'].apply(get_energy)
energy['Country'] = energy['Country'].str.slice_replace(-1,repl='')

这用最后一个元素替换了Country中的每个项目,但我不希望它在所有地方都替换。

0 个答案:

没有答案