代码之家  ›  专栏  ›  技术社区  ›  Prateek Narendra

将集合的数组转换为矩阵

  •  -2
  • Prateek Narendra  · 技术社区  · 6 年前

    我从文件中加载了一个数据集-

    np.array([(5.1, 3.5, 1.4, 0.2, 'Iris-setosa'),
              (4.9, 3., 1.4, 0.2, 'Iris-setosa'),
              (4.7, 3.2, 1.3, 0.2, 'Iris-setosa'),
              (4.6, 3.1, 1.5, 0.2, 'Iris-setosa'),
              (5., 3.6, 1.4, 0.2, 'Iris-setosa')])
    

    如何得到前4列的矩阵-

    np.array([[5.1, 3.5, 1.4, 0.2],
              [4.9, 3., 1.4, 0.2],
              [4.7, 3.2, 1.3, 0.2],
              [4.6, 3.1, 1.5, 0.2],
              [5., 3.6, 1.4, 0.2]])
    

    并将标签放入另一个数组中-

    np.array(['Iris-setosa', 'Iris-setosa', 'Iris-setosa', 'Iris-setosa',
              'Iris-setosa'])
    
    1 回复  |  直到 6 年前
        1
  •  0
  •   J. Doe    6 年前
    import pandas as pd
    
    data=np.array([(5.1, 3.5, 1.4, 0.2, 'Iris-setosa') ,
     (4.9, 3. , 1.4, 0.2, 'Iris-setosa'),
     (4.7, 3.2, 1.3, 0.2, 'Iris-setosa'),
     (4.6, 3.1, 1.5, 0.2, 'Iris-setosa'),
     (5. , 3.6, 1.4, 0.2, 'Iris-setosa')])
    
    df= pd.DataFrame(data)
    first4col = df.iloc[:,0:3]
    tags=df.iloc[:,4]