1

来自 RI 用于通过使用以下索引为矩阵分配一个简单的值：

> mat <- matrix(NA,2,3)
> mat
     [,1] [,2] [,3]
[1,]   NA   NA   NA
[2,]   NA   NA   NA
> mat[1,2] <- 12
> mat
     [,1] [,2] [,3]
[1,]   NA   12   NA
[2,]   NA   NA   NA
>

现在我必须在 Python 中运行一些脚本，并且还想在 2D 矩阵中收集值。矩阵应该有列名和行名，我也想通过索引访问它们。到目前为止，我想到了两种方法：

  • 使用 RPy 并按照我在 R 中习惯的方式进行操作，尽管我还不知道索引命令如何与 RPy 一起工作。

像：

import rpy2.robjects as robjects

names = ['A','B','C']
dimnames = robjects.r['list'](names,names) 
mat = robjects.r['matrix'](NA, nrow = len(names), ncol=len(names), dimnames = dimnames)

# and here how to index and assign e.g. the value 12 to the cell mat["A","B"]??
  • 或者使用 Python 本机 NumPy 数组，这可能是一个更清洁的解决方案。目前我只是不知道如何使用名称列表创建矩阵/数组以及如何使用索引分配值。

有什么建议么？

谢谢！！

4

2 回答 2

3

我认为 pandas 的DataFrame 数据类型可能适合您。

import pandas
matrix = pandas.DataFrame(index=['a','b','c'], columns=['one','two'])

返回

   one  two
a  NaN  NaN
b  NaN  NaN
c  NaN  NaN

您可以更改条目：matrix['one']['a']=1，以及其他熟悉的 numpy 操作（如上述链接中所述）。

于 2012-08-24T10:32:09.723 回答
0

您可以使用字典将行/列名称映射到索引。

names = ['A', 'B', 'C']
idx = range(0, len(names))
d = dict((key, value) for (key, value) in zip(names, idx))

然后用它来访问矩阵中的元素，

mat[d['A'], d['B']] = 12

据我从rpy2的文档中可以看出，这应该可行；如果不是，也许下面的语法更合适：

mat.rx(d('A'), d('B'))
于 2012-08-24T10:33:01.520 回答