从python访问矩阵的运算符函数

2024-05-23 14:33:48 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图访问G4旋转矩阵的矩阵元素

G4旋转矩阵类型定义为HEP旋转,如下所示 typedef CLHEP::HepRotation G4RotationMatrix

HepRotation的功能之一定义为

double HepRotation::operator() (int i, int j) const {
  if (i == 0) {
    if (j == 0) { return xx(); }
    if (j == 1) { return xy(); }  
    if (j == 2) { return xz(); }
  } else if (i == 1) {
    if (j == 0) { return yx(); }
    if (j == 1) { return yy(); }
    if (j == 2) { return yz(); }
  } else if (i == 2) {
    if (j == 0) { return zx(); }
    if (j == 1) { return zy(); }
    if (j == 2) { return zz(); }
  }
  std::cerr << "HepRotation     subscripting: bad indices "
       << "(" << i << "," << j << ")" << std::endl;
  return 0.0;

假设已经正确定义了它,以增强我如何以及如何用Python编写代码来访问矩阵元素(在我的例子中称为rot)


Tags: 元素returnif定义矩阵elsehepint