python - 使用python将 'repetitive' 2D数组转换为矩阵

Question

我有一个文本文件，其中包含以下形式的信息:

A  0
B  1
C  4
D  0
E  1
A  0
B  0
C  2
D  1
E  1
A  1
B  0
C  2
D  0
E  0
...

请注意，如果不计算 ABCDE 循环的总数(此处仅显示 3 个)，则无法得知它们。我想使用 Python 将其转换为以下形式的矩阵:

A  0 0 1 ...
B  1 0 0 ...
C  4 2 2 ...
D  0 1 0 ...
E  1 1 0 ...

我不确定进行此类转换的最佳方法是什么，有人作为 python 脚本执行此操作吗？ Numpy 或 Pandas 中是否有任何函数可以轻松做到这一点？或者我应该在没有 Numpy 或 Pandas 的情况下进行此操作？

非常感谢您的帮助!

Answer 1

Pandas 解决方案:

import pandas as pd
from pandas.compat import StringIO

temp=u"""
A  0
B  1
C  4
D  0
E  1
A  0
B  0
C  2
D  1
E  1
A  1
B  0
C  2
D  0
E  0"""
#after testing replace StringIO(temp) to filename
df = pd.read_csv(StringIO(temp), sep="\s+", header=None)

print (df)
    0  1
0   A  0
1   B  1
2   C  4
3   D  0
4   E  1
5   A  0
6   B  0
7   C  2
8   D  1
9   E  1
10  A  1
11  B  0
12  C  2
13  D  0
14  E  0

df = pd.pivot(index=df[0], columns=df.groupby(0).cumcount(), values=df[1])
print (df)
   0  1  2
0         
A  0  0  1
B  1  0  0
C  4  2  2
D  0  1  0
E  1  1  0