创建按字符串索引的python 2d数组

Question

（我不太擅长使用p​​ython，因此如果这是一个愚蠢的问题，请原谅我）

所以我想创建一个表示这种情况的数据结构-

      word1   word2
word3   1      2
word4   3      4

现在我已经尝试做这样的事情-

self.table = [][]

但这给了我一个无效的语法错误（我想是因为我还没有初始化数组？）。 但是，即使我要使用它，也将无法使用它，因为我不知道我的x和y尺寸有多大（似乎会有索引异常数组）。

我应该使用双重字典吗？ 我应该使用什么？

Answer 1

这个想法是创建一个字典，将字符串映射到索引。 在下面的代码中，有一个小类可以重载'[]'运算符：

class ArrayStrIdx:
    """ 2D Array with string as Indeces

    Usage Example::

        AA = ArrayStrIdx()
        AA["word1", "word3"] = 99
        print(AA["word2", "word5"])
    """

    cols ="word1 word2"
    rows ="word3 word4 word5"

    dd = [[10,11,12],[20,21,22]]  # data

    def __init__(self):
        """ Constructor builds dicts for indexes"""
        self.ri = dict([(w,i) for i,w in enumerate(self.rows.split())])
        self.ci = dict([(w,i) for i,w in enumerate(self.cols.split())])

    def _parsekey(self, key):
        """ Convert index key to indeces for ``self.dd`` """
        w0, w1 = key[0], key[1]
        return self.ci[w0], self.ri[w1]

    def __getitem__(self, key):
        """ overload [] operator - get() method"""
        i0, i1 = self._parsekey(key)
        return self.dd[i0][i1]

    def __setitem__(self, key, value):
        """ overload [] operator -  set() method """
        i0, i1 = self._parsekey(key)
        self.dd[i0][i1] = value

更新：扩展答案以允许类似AA["word1", "word3"] = 23 。

Answer 2

也许您可以尝试使用初始化表

self.table = {r : { c : 0 for c in ['word1', 'word2']} for r in ['word3', 'word4']}

然后您可以通过以下方式访问每个职位

self.table['word3']['word1'] = 2

Answer 3

Python没有创建二维矩阵的现成指令，但是使用列表推导很容易做到这一点：

matrix = [[0] * cols for i in range(rows)]

然后您可以使用

matrix[row][col] += 42

假设rows=10 ， cols=20则row索引必须从0到9， col索引必须从0到19。 例如

matrix[-1][-1] = 99

将最后一个单元格设置为99

Answer 4

如果您不反对使用外部库，则可以签出

熊猫数据框