python - 如何将数据框列拆分为 Pandas 中的另外两列？

Question

我试图将名为“变量”的列拆分为另外两列“类型”和“参数”

  BatchNumber PhaseNumber SiteID   variable  Values
0  4552694035       0020B      2  min_tempC    27.0
1  4552694035     OverAll      2  max_tempF    24.0

我尝试使用下面的代码

weatherData = weatherData['variable'].str.split('_', 1)

但没有得到预期的结果。预期结果如下。

  BatchNumber PhaseNumber SiteID   variable  Values     Type    Parameter
0  4552694035       0020B      2  min_tempC    27.0     min     tempC
1  4552694035     OverAll      2  max_tempF    24.0     max     tempF

任何人都知道..如何得到它？

Answer 1

使用DataFrame.pop用于使用 split 提取列和 DataFrame 的参数 expand=True:

weatherData[['Type','Parameter']]=weatherData.pop('variable').str.split('_', 1, expand=True)
print (weatherData)
   BatchNumber PhaseNumber  SiteID  Values Type Parameter
0   4552694035       0020B       2    27.0  min     tempC
1   4552694035     OverAll       2    24.0  max     tempF

如果还想删除原始列 pop:

weatherData[['Type','Parameter']] = weatherData['variable'].str.split('_', 1, expand=True)
print (weatherData)
   BatchNumber PhaseNumber  SiteID   variable  Values Type Parameter
0   4552694035       0020B       2  min_tempC    27.0  min     tempC
1   4552694035     OverAll       2  max_tempF    24.0  max     tempF