我有一个文本文件,其中包含以下形式的信息:
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 的情况下进行此操作?
非常感谢您的帮助!
最佳答案
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
关于python - 使用python将 'repetitive' 2D数组转换为矩阵,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41137343/