Python筛选字符串中数字并重新排列

最新推荐文章于 2023-06-16 10:45:46 发布

原创最新推荐文章于 2023-06-16 10:45:46 发布 · 2.4k 阅读

1 ·

CC 4.0 BY-SA版权

编程基础同时被 2 个专栏收录

32 篇文章

订阅专栏

python

20 篇文章

订阅专栏

本文介绍了一种从不规则列表中提取数字并将其重塑为矩阵的方法。通过使用Python的字符串处理技巧及NumPy库，将嵌套列表中的所有数字提取出来，并最终转换成固定维度的矩阵形式。

data是一个列表，要求提取出其中的数字，并按原规律排列成矩阵

>>> data = [[1, 2, [3, 4]], 5, 6, [7, 8], [1, 3, [6, 7]]]
>>> data
[[1, 2, [3, 4]], 5, 6, [7, 8], [1, 3, [6, 7]]]
>>> str_data = str(data)
>>> filter_digit = []
>>> for i in str_data:
...     if i.isdigit():
...             filter_digit.append(i)
...
>>> filter_digit
['1', '2', '3', '4', '5', '6', '7', '8', '1', '3', '6', '7']
>>> import numpy as np
>>>> arr_data = np.array(filter_digit)
>>> arr_data = np.array(filter_digit).reshape(3,4)
>>> arr_data
array([['1', '2', '3', '4'],
       ['5', '6', '7', '8'],
       ['1', '3', '6', '7']],
      dtype='<U1')