python - 有没有办法使用 pandas 根据长度在字符串的字符之间插入空格？

Question

我一直在尝试编写一个函数，根据列中字符串的长度插入空格。

我有一个包含两列的数据框，一列包含邮政编码，另一列包含这些邮政编码的长度。如下图所示:

    new_pstl_cd length
  1 SS55HA       6
  2 BD108EG      7
  3 LS15HU       6
  4 W19PX        5

我想插入一个空格，使该列变成

   new_pstl_cd  
 1 SS5 5HA
 2 BD10 8EG         
 3 LS1 5HU
 4 W1 9PX

我尝试了以下代码但没有成功:

 def insert_space(charachter):
      if postcode_test['length'] == 6:
          return (postcode_test['new_pstl_cd'].str[0:3]+ charachter + postcode_test['new_pstl_cd'].str[3:])
      if postcode_test['length'] == 5:
          return (postcode_test['new_pstl_cd'].str[0:2]+ charachter + postcode_test['new_pstl_cd'].str[3:])
      else:
           return (postcode_test['new_pstl_cd'].str[0:4]+ charachter + postcode_test['new_pstl_cd'].str[4:])

我如何使用长度来编写一个函数来执行此操作？请注意，在这种情况下，始终会分隔最后 3 个字符

Answer 1

你甚至不需要长度:

df['new_pstl_cd'] = df.new_pstl_cd.str[:-3] + ' ' + df.new_pstl_cd.str[-3:]

输出:

  new_pstl_cd  length
1     SS5 5HA       6
2    BD10 8EG       7
3     LS1 5HU       6
4      W1 9PX       5