解包PLSRegression生成的numpy数组提取坐标用于二维散点图可视化

你可以直接通过numpy的数组切片操作实现按列提取：

• 你贴出的PLS计算结果外层是元组包裹numpy数组的格式，需要先提取元组内的数组对象，再按列切片
• 切片规则[:, 列索引]表示选中所有行的指定列，索引从0开始计数，第一列对应索引0，第二列对应索引1
• 如果不需要Python列表格式，去掉.tolist()即可保留numpy数组格式，也可以直接用于绘制散点图

代码示例 #

import numpy as np

# 假设your_pls_result是你调用PLSRegression得到的原始返回结果
# 第一步：从元组中提取numpy数组，若返回结果本身就是数组可跳过这一步
pls_arr = your_pls_result[0]

# 第二步：按列提取坐标
X_1 = pls_arr[:, 0].tolist()
X_2 = pls_arr[:, 1].tolist()

运行代码后得到的X_1、X_2完全符合你给出的预期结果，可直接传入matplotlib等绘图库使用。

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