根据指数值查找矩阵数据

时间:2018-07-06 15:52:38

标签: python arrays matrix lookup

我需要创建一个数组(列表)或矩阵,用作key:value对,并允许我查找有关公司的信息。

Python似乎不支持2d数组,所以我改用Matrix。

Matrix = {}

Matrix[1,1] = '1'
Matrix[1,2] = 'Dog Company'
Matrix[1,3] = '0'
Matrix[1,4] = '../../img/dogcompany_logo.png'

我的python脚本使用此数据根据输入的公司名称构建静态HTML文件。

我需要编写一个函数,该函数返回指定公司名称的行索引,并使用该索引获取该行中另一列的值。

每个公司都有一个唯一的名称,因此只能有一个包含指定公司名称的行索引。

例如:

[pseudo-code]
def rowIndex(companyName):

    for x in range(1, Matrix.length):
        if Matrix[x,2] = companyName:
            return x

myIndex = rowIndex('Dog Company')
logoURL = Matrix[myIndex,4]

如果有人能帮助我使其正常工作或提供更好的解决方案,我将不胜感激。

1 个答案:

答案 0 :(得分:0)

我不确定您要存储什么,但是您可以使用它。但是,如果要长期持久,请使用数据库或json文件

from collections import namedtuple

Company = namedtuple('Company', 'info1 info2')
companies = {"Dog Company" : Company("Data for info1", "Data for info2")}

def get_company(company):
   if company in companies:
      return companies[company]
   return None

get_company("Dog Company")