python - 在 numpy ndarray 中查找列表的长度

Question

我有一个形状为(100,)的numpy数组arr

  arr.shape()  #(100,)

此数组中有 100 个列表。列表是没有空格的字符串，长度大约为 30

'TBTBBBBBBBBBEBEBEBEBDLKDJFDFIKKKKK'

这个数组有一个“矩阵式”的形状。我怎样才能 (a) 告诉这个数组中每个列表的长度或 (b) 重新格式化这个数组，以便 arr.shape() 给我两个维度，即 arr。 shape() 给出 (100,30)？

另外，可能不是每个列表的长度都相同(有些可能是 28，而不是 30):

如何检查此行为？在这种情况下，numpy.shape() 会输出什么？

Answer 1

您可以使用列表理解来确定数组中每个字符串的长度:

arr = np.array(['aasdfads', 'asfdads', 'fdfsdfaf'])
>>> [len(i) for i in arr]
[8, 7, 8]

或者取最大值:

>>> max([len(i) for i in arr])
8