np.newaxis使用心得

最新推荐文章于 2023-02-13 19:09:54 发布

又要起名字了

最新推荐文章于 2023-02-13 19:09:54 发布

阅读量4.1k

点赞数 19

分类专栏： python小技巧文章标签： np.newaxis np.hstack

本文链接：https://blog.csdn.net/weixin_44530236/article/details/86490275

版权

本文通过实例解析np.newaxis在numpy操作中的作用，详细介绍了如何利用np.newaxis增加新的维度，包括在不同位置添加时对数据形状的影响。通过示例展示了在机器学习中处理数据集时，如何解决因维度不匹配导致的问题，如在融合sklearn的diabetes数据集和iris数据集时的技巧。

摘要由CSDN通过智能技术生成

最近对着sklearn文档练习机器学习（Machine Learning, ML），代码中遇到一个问题np.newaxis。

import matplotlib.pyplot as plt
import numpy as np
from sklearn import datasets, linear_model
from sklearn.metrics import mean_squared_error, r2_score

# Load the diabetes dataset
diabetes = datasets.load_diabetes()

# Use only one feature
diabetes_X = diabetes.data[:, np.newaxis, 2]

经查看文档，发现np.newaxis的作用就是增加一个新的维度，下面结合diabetes.data数据集加以说明。

#原始数据集维度
data = diabetes.data
data.shape