Python转移矩阵

Question

我有个清单是这样的：

2，1，3，1，2，3，1，2，2，2

我想要的是一个过渡矩阵，它显示的顺序如下：

• 1跟1的频率是多少？
• 1跟2的频率是多少？
• 1跟3的频率是多少？
• 2跟1的频率是多少？
• 2跟2的频率是多少？
• 2跟3的频率是多少？

等等..。

((0,2,1)，(1,2,1)，(2,0,0))

有没有预先制作好的模块去拿这个？

Answer 1

我不知道是否有模块，但我会使用以下代码，这是很容易泛化的：

import numpy as np
from collections import Counter
a = [2, 1, 3, 1, 2, 3, 1, 2, 2, 2]
b = np.zeros((3,3))
for (x,y), c in Counter(zip(a, a[1:])).iteritems():
    b[x-1,y-1] = c
print b
array([[ 0.,  2.,  1.],
       [ 1.,  2.,  1.],
       [ 2.,  0.,  0.]])

没有安装numpy：

b = [[0 for _ in xrange(3)] for _ in xrange(3)]
for (x,y), c in Counter(zip(a, a[1:])).iteritems():
    b[x-1][y-1] = c

print b
[[0, 2, 1], [1, 2, 1], [2, 0, 0]]

如果需要的话，详细介绍一下正在发生的事情：

1. zip(a, a[1:])获取所有的连续数字对。
2. Counter计算每对出现多少次
3. for循环简单地将Counter生成的字典转换为所请求的矩阵/列表。