Python在numpy数组中拆分字符串

Question

我有一个这样的numpy数组

Input

array([['ATS1, ATS2', 'P_CD'],
       ['ATS1,ATS2,ATS3', 'C_CD']], dtype=object)

我想转换这个numpy数组，如下所述

Expected output

array([['ATS1' , 'ATS2', 'P_CD'],
       ['ATS1','ATS2','ATS3', 'C_CD']], dtype=object)

正如您在上面所注意到的，我想用delimeter拆分字符串，并将其作为一个单独的条目

关于如何使用python实现这一目标，有什么建议吗

Answer 1

您可以使用re.split和join
这只是更改类型，但这会导致列表的numpy数组，因为内部子列表的长度可以是可变的，因此它们不会是numpy数组类型。

import numpy as np
import re

arr = np.array([['ATS1, ATS2', 'P_CD'],
       ['ATS1,ATS2,ATS3', 'C_CD']], dtype=object)
arr = np.array([re.split('[-,]','-'.join(ele)) for ele in arr] ,dtype=object)
print(arr)