使用pandas怎么修改DataFrame列名

這篇文章將為大家詳細(xì)講解有關(guān)使用pandas怎么修改DataFrame列名，文章內(nèi)容質(zhì)量較高，因此小編分享給大家做個(gè)參考，希望大家閱讀完這篇文章后對(duì)相關(guān)知識(shí)有一定的了解。

提出問(wèn)題

存在一個(gè)名為dataset的DataFrame

>>> dataset.columns
Index(['age', 'job', 'marital', 'education', 'default', 'housing', 'loan',
    'contact', 'month', 'day_of_week', 'duration', 'campaign', 'pdays',
    'previous', 'poutcome', 'emp.var.rate', 'cons.price.idx',
    'cons.conf.idx', 'euribor3m', 'nr.employed', 'y'],
   dtype='object')

現(xiàn)在, 我要將其columns名字改為:

>>> new_columns
Index(['age_0', 'job_1', 'marital_2', 'education_3', 'default_4', 'housing_5',
    'loan_6', 'contact_7', 'month_8', 'day_of_week_9', 'duration_10',
    'campaign_11', 'pdays_12', 'previous_13', 'poutcome_14',
    'emp.var.rate_15', 'cons.price.idx_16', 'cons.conf.idx_17',
    'euribor3m_18', 'nr.employed_19', 'y_20'],
   dtype='object')

該如何操作？

解決

一.通過(guò)DataFrame.columns類的自身屬性修改:

1.無(wú)腦賦值直接修改

>>> # 先解決`new_columns`的推導(dǎo)問(wèn)題
>>> # 列表推導(dǎo)
>>> new_columns_list = [column_str+'_'+str(i) for i ,column_str in enumerate(dataset.columns)]
>>> # 類型轉(zhuǎn)換
>>> new_columns = pd.core.indexes.base.Index(new_columns_list)
>>> dataset.columns = new_columns

2.通過(guò).map(mapper, na_action=None)函數(shù)來(lái)修改

>>> # 注:mapper 多運(yùn)用 lambda 表達(dá)式
>>> # 但我似乎沒有找到在 lambda 表達(dá)式中改變兩個(gè)值的方法
>>> # 所以只能蹩腳地用一個(gè)全局變量i, 和映射函數(shù)mapper()
>>> # 希望大家能幫我找到方法

>>> i = 0
>>> def mapper(x): # 映射函數(shù)即 mapper
  global i
  x += '_' + str(i)
  i += 1
  return x
>>> dataset.columns.map(mapper)

3.參考博客用到了DataFrame.columns.str對(duì)象

用help(DataFrame.columns.str)翻遍了文檔，

也沒能找到可以被我拿來(lái)套用的方法, 想著抽時(shí)間把這段文檔翻譯一下

二.通過(guò)DataFrame.rename()函數(shù)來(lái)修改

1.暴力字典法(好處：可以只修改特定的列)

>>> # 此處先用字典推導(dǎo)法
>>> new_dict = {
  key:key+'_'+str(i)
  for i, key in enumerate(dataset.columns)
  }
>>> dataset.rename(columns=new_dict, inplace=True)

2.映射修改法

>>> # 原博文依然用到了 lambda 表達(dá)式
>>> # 我就再生搬硬套一次, 把上面的復(fù)制過(guò)來(lái)
>>> # 蹩腳地用一個(gè)全局變量i, 和映射函數(shù)mapper()

>>> i = 0
>>> def mapper(x): # 映射函數(shù)即 mapper
  global i
  x += '_' + str(i)
  i += 1
  return x
dataset.rename(columns=mapper, inplace=True)

關(guān)于使用pandas怎么修改DataFrame列名就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，可以學(xué)到更多知識(shí)。如果覺得文章不錯(cuò)，可以把它分享出去讓更多的人看到。