如何在 Python 中使用 uproot 读取 TMatrixT？

问题描述

我有一个 *.root 文件，我正在尝试使用 uproot (uproot4) 读入 Python。它看起来像这样：

>>> data = up.open('file.root')
>>> data.keys()
['ring_sums;1','tpc_multiplicity;1','impact_parameter;1']
>>> data['ring_sums']
<TMatrixT<double> (version 4) at 0x0205b219d748>

如何读取 TMatrixT 类型？使用 .keys()、.values() 和 .items() 不起作用。

>>> data['ring_sums'].values()
Traceback (most recent call last):
  File "<stdin>",line 1,in <module>
AttributeError: 'Model_TMatrixT_3c_double_3e__v4' object has no attribute 'values'

将数据读入笨拙的数组、pandas 数据框或 numpy 数组的正确方法是什么？

解决方法

TMatrixT 是一个 Uproot 没有任何专门代码的类（还没有？），但无论如何它都可以被读取，因为它的“流媒体”（反序列化指令）包含在 ROOT 文件中。因此，它以一种相当通用的方式呈现。

您可以在 all_members (dict) 中查找对象的所有成员数据，也可以使用 member("memberName") 方法仅提取一个。因此，例如，尝试

data['ring_sums'].member("fElements")

因为 TMatrixT::fElements 似乎是相关的。这些可能会以未成形的 NumPy 数组（或 uproot.models.TArray，它是 NumPy 数组的子类）的形式出现。您可能需要将其 reshape 转化为正确的形式，并且可能会在其他某个成员（可能属于超类的成员）中找到正确的形状。

如果您找到使用 TMatrixT 对象的好方法并希望它自动化，也许在 values 方法中，请将其添加为 pull request 或将其描述为 { {3}} 用于 Uproot（一种新的“行为”）。