如何基于占位符填充矩阵：统一替换字母为对应0-9整数？

现有一个以字母作为占位符的NumPy矩阵，需要将这些字母替换为0-9的整数，要求相同字母必须对应相同整数；每行的字符序列组成一个数字，该数字需满足自定义的rule1函数验证条件。

原矩阵如下：

import numpy as np
matrix = np.array([['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'D', 'D', 'D'],
                   ['A', 'E', 'E', 'E', 'B', 'B', 'C', 'D', 'D', 'D', 'F'],
                   ['A', 'E', 'E', 'B', 'B', 'B', 'C', 'D', 'D', 'D', 'F']])

我尝试通过嵌套循环枚举数字组合，并用rule1验证每行数字，但这种写法无法泛化——一旦矩阵中的占位符数量变化，就得手动修改循环层数。而且不知道如何更高效地识别所有相同占位符并统一赋值，原尝试代码如下：

row1 = matrix[0]
i = 0
while i < 10:
    row1[0] = row1[1] = row1[2] = i
    j = 0
    while j < 10:
        row1[3] = row1[4] = row1[5] = j
        k = 0
        while k < 10:
            row1[6] = row1[7] = k
            y = 0
            while y < 10:
                row1[8] = row1[9] = row1[10] = y
                l = 0
                num = 0
                while l < 11:
                    num = num + pow(10, 10 - l) * int(row1[l])
                    l += 1
                if rule1(num):
                    return num
                y += 1
            k += 1
        j += 1
    i += 1

核心思路 #

• 提取唯一占位符：自动从矩阵中识别所有不同字母，避免手动枚举
• 生成数字映射组合：用itertools.product替代多层嵌套循环，实现代码泛化
• 批量替换与验证：将数字映射应用到整个矩阵，转换每行成数字后统一验证规则

实现代码 #

import numpy as np
from itertools import product

# 自定义验证规则示例（根据实际需求修改）
def rule1(num):
    # 示例：数字能被3整除
    return num % 3 == 0

# 原矩阵
matrix = np.array([['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'D', 'D', 'D'],
                   ['A', 'E', 'E', 'E', 'B', 'B', 'C', 'D', 'D', 'D', 'F'],
                   ['A', 'E', 'E', 'B', 'B', 'B', 'C', 'D', 'D', 'D', 'F']])

# 提取所有唯一占位符
unique_chars = np.unique(matrix)
char_count = len(unique_chars)

# 生成所有可能的数字组合（允许字母对应重复数字；若要求不同字母对应不同数字，改用permutations(range(10), char_count)）
for digits in product(range(10), repeat=char_count):
    # 创建字母到数字的映射字典
    char_to_num = dict(zip(unique_chars, digits))
    
    # 批量替换矩阵中的字母为数字
    num_matrix = np.vectorize(lambda c: char_to_num[c])(matrix)
    
    # 将每行转换为整数
    row_nums = []
    for row in num_matrix:
        num = 0
        for idx, digit in enumerate(row):
            num += digit * (10 ** (len(row) - 1 - idx))
        row_nums.append(num)
    
    # 验证所有行是否满足规则（可按需调整为验证特定行）
    if all(rule1(num) for num in row_nums):
        print("找到符合条件的映射：", char_to_num)
        print("对应数字：", row_nums)
        break

关键说明 #

• 泛化性：无论矩阵中有多少种占位符，代码都会自动识别并生成对应组合，无需修改循环结构
• 灵活调整：若要求不同字母必须对应不同数字，只需替换生成组合的方法为permutations(range(10), char_count)
• 效率优化：可提前添加约束过滤无效组合，比如检查每行首位数字是否为0，减少不必要的计算

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