将60k x 10阵列压成60k x 1

时间:2018-12-04 11:26:19

标签: python sparse-matrix reduction

我在mat文件(matlab文件)中有mnist数据。我已经加载了文件。 现在,我有60000x10的巨大稀疏矩阵表示标签。这是60K位数字,并且10列中只有一个具有等于1的条目,代表该数字。 例如

0000000001 = 9
1000000000 = 0
...

依此类推

需要将其压缩为60k x 1,以便一维现在代表数字的标签

9
0
...

它是一个numpy数组,所以我要为此去熊猫吗?

谢谢

0 个答案:

没有答案