可以使用Python中的numpy库来解决此问题。可以使用numpy.append()函数创建并填充新数组,并检查当前数组是否需要扩展其大小。
示例代码如下:
import numpy as np
假设当前数组为arr1,需要扩展的列数为2
arr1 = np.array([[1, 2, 3], [4, 5, 6]])
new_cols = 2
检查需要扩展的列数是否大于当前数组的列数
if new_cols > arr1.shape[1]:
# 计算新数组的大小
new_shape = (arr1.shape[0], new_cols - arr1.shape[1])
# 创建并填充新数组
new_arr = np.append(arr1, np.zeros(new_shape), axis=1)
# 打印新的数组
print(new_arr)
else:
# 列数不需要扩展,直接打印当前数组
print(arr1)
运行以上代码将输出扩展后的新数组。注意,需要指定axis参数来确定在哪个维度上添加元素。在本例中,我们在列维度上添加了元素。