Question

我被困（并且有点时间紧张），并希望得到一些帮助。这可能是一项简单的任务，但我似乎无法解决它......

我有一个矩阵，比如5乘5，另外还有一个行名称的起始列和文本文件中列的相同名称，如下所示：

b e a d c
b 0.0 0.1 0.3 0.2 0.5
e 0.1 0.0 0.4 0.9 0.3
a 0.3 0.4 0.0 0.7 0.6
d 0.2 0.9 0.7 0.0 0.1
c 0.5 0.3 0.6 0.1 0.0

我有多个文件具有相同的矩阵格式和大小，但名称的顺序不同。我需要一种方法来改变它们，所以它们都是一样的并保持0.0对角线。所以我要对列进行交换，我必须对行进行交换。

我一直在搜索，似乎NumPy可能会做我想要的但是我从来没有使用它或一般的数组。任何帮助是极大的赞赏！

简而言之：如何将文本文件放入数组中，然后我可以将行和列交换到所需的顺序？

Answer 1

我建议你使用熊猫：

from StringIO import StringIO
import pandas as pd
data = StringIO("""b e a d c
b 0.0 0.1 0.3 0.2 0.5
e 0.1 0.0 0.4 0.9 0.3
a 0.3 0.4 0.0 0.7 0.6
d 0.2 0.9 0.7 0.0 0.1
c 0.5 0.3 0.6 0.1 0.0
""")
df = pd.read_csv(data, sep=" ")
print df.sort_index().sort_index(axis=1)

输出：

     a    b    c    d    e
a  0.0  0.3  0.6  0.7  0.4
b  0.3  0.0  0.5  0.2  0.1
c  0.6  0.5  0.0  0.1  0.3
d  0.7  0.2  0.1  0.0  0.9
e  0.4  0.1  0.3  0.9  0.0

Answer 2

这是一个可怕的Numpy版本的开始（使用HYRY的答案......）

import numpy as np

with open("myfile", "r") as myfile:
    lines = myfile.read().split("\n")
    floats = [[float(item) for item in line.split()[1:]] for line in lines[1:]]
    floats_transposed = np.array(floats).transpose().tolist()

Answer 3

from copy import copy

f = open('input', 'r')
data = []
for line in f:
    row = line.rstrip().split(' ')
    data.append(row)

#collect labels, strip empty spaces
r = data.pop(0)
c = [row.pop(0) for row in data]
r.pop(0)

origrow, origcol = copy(r), copy(c)

r.sort()
c.sort()

newgrid = []
for row, rowtitle in enumerate(r):
    fromrow = origrow.index(rowtitle)
    newgrid.append(range(len(c)))
    for col, coltitle in enumerate(c):
        #We ask this len(row) times, so memoization
        #might matter on a large matrix
        fromcol = origcol.index(coltitle)
        newgrid[row][col] = data[fromrow][fromcol]

print "\t".join([''] + r)
clabel = c.__iter__()
for line in newgrid:
    print "\t".join([clabel.next()] + line)

重新排列矩阵与命名列/行python

3 个答案: