python - 在python中存储大量 bool 数据

Question

我需要存储稀疏矩阵数据。数据大小为 10^6 10^4 列。在每一列中，我都存储了一个由 0 组成的向量，除了少数为 true 的值。

然后我需要对每个矩阵中的列求和，并将每一行乘以一个标量。我尝试过字典，但当我需要求和和乘法时它们会失败。

你会用什么？

附言。 numpy.zeros 太小了

Answer 1

两个字典怎么样？假设这是矩阵(x for True):

   0  1  2  3  4  5  6  7
0  x     x        x 
1     x
2                       x
3              x
4
5
6        x        x
7

你只需要存储

rows = {0: [0, 2, 5], 1: [1], 2: [7], 3: [4], 6: [2, 5]}

你可以轻松地将其转换为

columns = {0: [0], 1: [1], 2: [0, 6], 4: [3], 5: [0, 6], 7: [2]}

使用类似的东西

columns = {}
for row in rows:
    for column in rows[row]:
        columns.setdefault(column, []).append(row)

然后对列求和 (sum(1 for x in column[2])) 或对行求和，然后将结果乘以任何你想要的值。