使用什么数据结构来构建不同大小的字符串列表的串联?
例如.,
a_list = ['h','i']
b_list = ['t','h','e','r','e']
c_list = ['fr', 'ie','nd']
理想结构:
my_structure = [ ['h','i'],
['t','h','e','r','e'],
['fr', 'ie','nd']
]
然后用’null’字符串填充它以在每个列表中获得相同的大小:
my_structure = [ ['h','i','null','null','null'],
['t','h','e','r','e'],
['fr', 'ie','nd','null', 'null']
]
最佳答案 你可以使用
itertools.zip_longest
:
import itertools
np.array(list(itertools.zip_longest(a_list, b_list, c_list, fillvalue='null'))).T
array([['h', 'i', 'null', 'null', 'null'],
['t', 'h', 'e', 'r', 'e'],
['fr', 'ie', 'nd', 'null', 'null']],
dtype='<U4')
编辑:根据您的注释,您希望向阵列添加新列表,创建要使用的列表列表可能更简单,并且您可以动态地追加到该列表:
a_list = ['h','i']
b_list = ['t','h','e','r','e']
c_list = ['fr', 'ie','nd']
my_list = [a_list, b_list, c_list]
my_arr = np.array(list(itertools.zip_longest(*my_list, fillvalue='null'))).T
>>> my_arr
array([['h', 'i', 'null', 'null', 'null'],
['t', 'h', 'e', 'r', 'e'],
['fr', 'ie', 'nd', 'null', 'null']],
dtype='<U4')
然后,您可以向my_list添加新列表:
d_list = ['x']
my_list.append(d_list)
my_arr = np.array(list(itertools.zip_longest(*my_list, fillvalue='null'))).T
>>> my_arr
array([['h', 'i', 'null', 'null', 'null'],
['t', 'h', 'e', 'r', 'e'],
['fr', 'ie', 'nd', 'null', 'null'],
['x', 'null', 'null', 'null', 'null']],
dtype='<U4')