python - 在 Pandas 中拆分一个字符串并将其加入旧数据

Question

我做的事情看似简单，但我想不通。

我有数据框，例如

City    State ZIP
Ames    IA    50011-3617
Ankeny  IA    50021

我想按 - 拆分邮政编码，并仅将第一个邮政编码保存在具有旧数据和新邮政编码的新数据框中。我尝试执行以下操作。

data_short_zip = data
df = data['ZIP'].str.split('-').str[0]
data_short_zip.join(df)

这不仅会引发错误，而且看起来不符合 Python 规范。有没有简单的方法可以做到这一点？

输出数据看起来像

City    State ZIP
Ames    IA    50011
Ankeny  IA    50021

Answer 1

您可以使用 str.split在分隔符上拆分，然后在结果上 str[0] 返回第一个拆分:

In [122]:
df['ZIP'] = df['ZIP'].str.split('-').str[0]
df

Out[122]:
     City State    ZIP
0    Ames    IA  50011
1  Ankeny    IA  50021