python - 剥离 numpy 数组中的空格

Question

我环顾四周，找不到一种简单/单行的方法来去除 numpy 数组中的空格::

print(type(p))
print(p)

<class 'numpy.ndarray'>
[{' SPL', 'GPU', 'bcc'} {'ANZ ', 'ROI'} {'ANZ', 'bcc'}
 {'GPU', '  ANZ   ', 'bcc'} {'bcc ', ' SPL'}]

到::

[{'SPL', 'GPU', 'bcc'} {'ANZ', 'ROI'} {'ANZ', 'bcc'}
 {'GPU', 'ANZ', 'bcc'} {'bcc', 'SPL'}]

尝试

np.char.strip(p)

结果

TypeError: string operation on non-string array

那么，这是变相的列表吗？

一点帮助？？!

Answer 1

你有一组集合，所以迭代它们。使用剥离的元素创建一个新集合，并创建一个新数组:

p = np.array([{item.strip() for item in s} for s in p])

这使用集合理解来创建带有剥离元素的新集合。原始数组在列表推导中迭代，然后被送入 np.array() 以创建一个新数组。