1

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

我有一个矩阵，比如 5 x 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 可能会做我想做的事，但我从来没有使用过它或一般的数组。任何帮助是极大的赞赏！

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

4

3 回答 3

4

我建议你使用熊猫：

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
于 2013-03-16T00:05:57.047 回答
0

这是一个可怕的 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()
于 2013-03-16T00:09:15.497 回答
0 
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)
于 2013-03-16T00:30:46.440 回答