python 将列表中的每个一维数组填充到同样长度
时间: 2024-03-03 09:53:29 浏览: 36
可以使用Python中的内置函数zip()和itertools.zip_longest()来实现将列表中的每个一维数组填充到同样长度的操作。
方法一:使用zip()函数
zip()函数可以将多个可迭代对象(如列表、元组等)中对应的元素打包成一个个元组,返回一个zip对象,可以通过list()函数将其转换为列表。
下面是将列表中的每个一维数组填充到同样长度的示例代码:
```python
my_list = [[1, 2], [3, 4, 5], [6, 7, 8, 9]]
# 获取列表中最长的一维数组的长度
max_len = max([len(l) for l in my_list])
# 使用zip函数将每个一维数组填充到同样长度
new_list = [list(x) + [None] * (max_len - len(x)) for x in zip(*my_list)]
print(new_list)
```
输出结果为:
```
[[1, 3, 6], [2, 4, 7], [None, 5, 8], [None, None, 9]]
```
方法二:使用itertools.zip_longest()函数
itertools.zip_longest()函数与zip()函数类似,不同的是它可以添加填充值,将不足长度的部分填充到指定长度。
下面是将列表中的每个一维数组填充到同样长度的示例代码:
```python
import itertools
my_list = [[1, 2], [3, 4, 5], [6, 7, 8, 9]]
# 获取列表中最长的一维数组的长度
max_len = max([len(l) for l in my_list])
# 使用itertools.zip_longest()函数将每个一维数组填充到同样长度
new_list = [list(x) for x in itertools.zip_longest(*my_list, fillvalue=None)]
new_list = [l + [None] * (max_len - len(l)) for l in new_list]
print(new_list)
```
输出结果为:
```
[[1, 3, 6], [2, 4, 7], [None, 5, 8], [None, None, 9]]
```
需要注意的是,以上两种方法都是将一维数组填充到最长数组的长度,如果需要填充到指定长度,可以将max_len替换为指定长度即可。