Python转换矩阵

时间:2014-08-12 16:29:16

标签: python transitions

我的列表看起来像这样:

[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))

有预制模块吗?

1 个答案:

答案 0 :(得分:13)

我不知道是否有模块,但我会使用此代码,这很容易概括:

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生成转换为您请求的矩阵/列表列表