如何从StepFunction数组中提取指定位置的y值？

首先将预测结果存入变量，然后通过索引直接访问对应StepFunction对象的y属性即可，注意Python数组是0索引，你要的第3个值对应的索引为2：

# 获取预测得到的StepFunction数组
survival_functions = estimator.predict_survival_function(data_x_numeric)

# 提取第一个用户y数组的第3个值
first_user_3rd_value = survival_functions[0].y[2]

# 提取第二个用户y数组的第3个值
second_user_3rd_value = survival_functions[1].y[2]

# 验证结果
print(first_user_3rd_value)   # 输出 0.67663527
print(second_user_3rd_value)  # 输出 0.86147605

如果StepFunction的y属性是numpy数组、普通列表这类可索引序列，以上代码直接生效；如果是其他类型，只要支持索引访问，写法逻辑一致。

内容的提问来源于stack exchange，提问作者Gopi S