ray.rllib.policy.sample_batch.SampleBatch.columns

SampleBatch.columns(keys: List[str]) → List[any]

返回指定列中的批处理数据列表。

参数:

keys – 要返回数据的列名列表。

返回:

按 keys 中列名的顺序排列的数据项列表。

from ray.rllib.policy.sample_batch import SampleBatch
batch = SampleBatch({"a": [1], "b": [2], "c": [3]})
print(batch.columns(["a", "b"]))

[[1], [2]]