Question

我正在编写将列表列表转换为矩阵的代码。此功能应可扩展到更大的列表，我的输入仅是为了使问题更易于解决。我还是个初学者，所以我需要一点帮助=）

-输入：在示例输入中，list1是水果和颜色对的数据集：

list1 = [[["apple", "red"], " 1 "],[["apple", "yellow"], " 1 "], [["apple", "green"], " 1 "]]
list1 += [[["lemon", "red"], " 0 "], [["lemon", "yellow"], " 1 "], [["lemon", "green"], " 0 "]]
list1 += [[["pear ", "red"], " 0 "], [["pear", "yellow"], " 0 "], [["pear", "green"], " 1 "]]

-所需的输出：

['', 'apple', 'lemon', 'pear']
['red', ' 1 ', 0, 0]
['yellow', 0, ' 1 ', 0]
['green', 1, 0, ' 1 ']

-我的输出

['///', 'apple', 'lemon', 'pear']
['red', ' 1 ', 0, 0]
['yellow', 0, ' 1 ', 0]
['green', 0, 0, ' 0 ']

------我的尝试：-----

# create empty matrix
matrix=[]
for row in range(4):
    new_row =[]
    for col in  range(4):
        new_row.append(0)   #if starting    all-0
    matrix.append(new_row)

    # add object names
    names = ["///", "apple", "lemon", "pear"]
    color = ["///", "red", "yellow", "green"]
    color = color[::-1]
    matrix[0] = names
    for row in matrix:
        row[0] = color.pop()

    second_obj = 0
    for row in range(4):
        for col in range(4):
            if list[row][0][0] == matrix[0][col]:
                while list[row][0][1] != matrix[second_obj][0]:
                    second_obj += 1
                else:
                    if matrix[row][row] != "///":
                        matrix[row][row] = list[row][1] 
                    second_obj = 0

Answer 1

编写一个手动循环来搜索列表会使事情变得过于复杂。编写两个循环以搜索两个字符串列表，并尝试将它们混合在一起，同时还循环索引其他内容……这也难怪您会感到困惑。

让我们废弃它，而使用一些词典：

columns = {'apple': 1, 'lemon': 2, 'pear': 3}
rows = {'red': 1, 'yellow': 2, 'green': 3}

现在，如果您想知道要放入哪个矩阵元素，则没有循环，只有两个字典查找：

>>> (colname, rowname), value = [["apple", "red"], " 1 "]
>>> columns[colname]
1
>>> rows[rowname]
1

所以，现在我们要做的就是从一个空矩阵开始：

matrix = [
    ['///', 'apple', 'lemon', 'pear'],
    ['red', 0, 0, 0],
    ['yellow', 0, 0, 0],
    ['green', 0, 0, 0]]

…遍历元素：

for (colname, rowname), value in list1:

…查找列和行：

    col = columns[colname]
    row = rows[rowname]

…并存储数字：

    matrix[row][col] = value

仅此而已。

好吧，差不多。您的数据有问题，其中一个字符串是'pear '，而不是'pear'。如果这是数据中的错误，则可以修复该错误。如果您的代码应该处理该问题，则必须决定如何处理它。一个明显的选择是从字符串中去除多余的空格：

    col = columns[colname.strip()]
    row = rows[rowname.strip()]

如果您不预先知道所有标签，需要以编程方式找到它们怎么办？

您可以在主列表之前再进行一次遍历，以拉出所有唯一的行和列名称。例如：

rows, columns = {}, {}
for (colname, rowname), value in list1:
    if rowname not in rows:
        next_unused_index = len(rows) + 1
        rows[rowname] = next_unused_index
    if colname not in columns:
        next_unused_index = len(columns) + 1
        columns[colname] = next_unused_index

现在，要构建矩阵，您需要根据这两个字典来构建。如果您使用的是python 3.7，则可以依靠dict顺序正确的事实，但是不依赖于dict可能更清楚。让我们先构建一个空矩阵：

matrix = [[0 for _ in range(len(columns)+1)] 
          for _ in range(len(rows)+1)]

…然后填写标题：

matrix[0][0] = '///'
for rowname, row in rows.items():
    matrix[row][0] = rowname
for colname, column in columns.items():
    matrix[0][column] = colname

…，然后您可以运行与之前相同的代码来填写值。

将数据列表转换成python矩阵（内部检查）

1 个答案: