如何利用两个数组索引构建矩阵？解决TypeError报错问题

看起来你在构建组合矩阵的时候踩了个常见的小坑，我来帮你梳理问题并解决～

首先，你的代码报错主要有两个核心问题：

• 你用生成器表达式直接赋值给com_total，这会覆盖你之前初始化的零矩阵，生成器对象和数组/矩阵类型不兼容，自然会抛出TypeError；
• 访问矩阵元素的语法错误：com_total(i,j)是函数调用的写法，数组/矩阵的元素访问应该用方括号索引，比如com_total[i][j]（普通列表）或com_total[i,j]（numpy数组）。

下面分两种常见场景给出具体修复方案：

场景1：使用普通Python列表构建矩阵 #

如果你的array_a和array_b是普通Python列表，比如array_a = [1, 2, 3]，array_b = [4, 5, 6]，可以用嵌套循环逐个赋值：

# 初始化零矩阵：行数=len(array_a)，列数=len(array_b)
com_total = [[0 for _ in range(len(array_b))] for _ in range(len(array_a))]

# 嵌套循环赋值（这里假设你想让com_total[i][j] = array_a[i] 和 array_b[j]的组合，比如相加，你可以替换成自己的逻辑）
for i in range(len(array_a)):
    for j in range(len(array_b)):
        com_total[i][j] = array_a[i] + array_b[j]  # 替换成你需要的赋值规则，比如直接用array_a[i]或者array_b[j]

场景2：使用NumPy构建矩阵（更高效） #

如果是处理数值矩阵，推荐用NumPy，它的广播机制可以不用写嵌套循环，代码更简洁高效：

import numpy as np

array_a = np.array([1, 2, 3])
array_b = np.array([4, 5, 6])

# 利用广播将array_a转为列向量，再和array_b进行运算（这里以相加为例，按需替换）
com_total = array_a[:, np.newaxis] + array_b

# 如果只是想把array_a的元素按行重复，或者array_b按列重复，也可以这样：
# com_total = np.tile(array_a[:, np.newaxis], (1, len(array_b)))  # 每行都是array_a的元素重复列数次数
# com_total = np.tile(array_b, (len(array_a), 1))  # 每列都是array_b的元素重复行数次数

为什么你的原代码不行？ #

你写的com_total = (com_total(i,j) for i in array_a for j in array_b)是创建了一个生成器对象，而不是修改原来的零矩阵。生成器本身不能直接作为矩阵使用，后续对它进行矩阵操作时就会因为类型不匹配报错。

内容的提问来源于stack exchange，提问作者Human Error