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

Question

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

-input：在示例输入中，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] = rownamefor colname, column in columns.items():    matrix[0][column] = colname…然后您可以运行与之前相同的代码来填写值。