如何在Python/Sage中将二进制矩阵转换为字符串/符号矩阵？

问题描述 #

现有一个由0和1组成的Sage矩阵：

m = [
    [1, 0, 1],
    [1, 1, 1],
    [0, 0, 1]
]

需要将其转换为符号/字符串形式，最终只需得到每行的字符串即可，期望结果如下：

"A 0 C"
"A B C"
"0 0 C"

尝试过通过矩阵与字符串向量相乘实现替换，但无法达到预期效果；创建字符串向量Mlet=vector(["A","B","C"])时出现报错，希望找到比逐元素遍历判断更简洁的实现方式。

解决方案 #

方法1：利用Sage矩阵的apply_map方法 #

Sage矩阵的apply_map方法可以对每个元素进行自定义映射，同时获取元素的位置索引，刚好能根据列索引匹配对应符号：

# 定义原矩阵
m = matrix([[1,0,1], [1,1,1], [0,0,1]])
# 定义对应列的符号列表
symbols = ['A', 'B', 'C']

# 转换为符号/字符串矩阵
m_str = m.apply_map(lambda val, pos: symbols[pos[1]] if val == 1 else '0')

# 转为每行的字符串
row_strings = [' '.join(row) for row in m_str]

执行后row_strings即为目标结果：["A 0 C", "A B C", "0 0 C"]

方法2：Python列表推导式（简洁高效） #

直接通过列表推导对每行元素进行替换，写法更简洁，Sage完全兼容这种Python式写法：

m = matrix([[1,0,1], [1,1,1], [0,0,1]])
symbols = ['A', 'B', 'C']

# 一行式生成所有行的字符串
row_strings = [' '.join([symbols[j] if val == 1 else '0' for j, val in enumerate(row)]) for row in m]

补充说明 #

• 创建字符串向量报错是因为Sage的默认vector构造器默认使用数值环（如整数环），无法直接存储字符串。实际上我们不需要将符号转为Sage向量，直接用Python列表即可满足需求。
• 矩阵乘法的思路不可行，因为矩阵乘法是求和运算，会将同一行中1对应的符号相加（比如第一行1*A + 0*B +1*C会得到A+C），而无法保留每个位置的独立符号/0。

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