logo
Loading...

合併list問題 - Cupoy

import numpy as npimport pandas as pdX = np.arange...

ml100-4,ml100-4-d34

合併list問題

2020/04/05 01:57 PM
機器學習新手論壇
Shih Cheng Chen
觀看數:0
回答數:2
收藏數:0
ml100-4
ml100-4-d34

import numpy as np
import pandas as pd
= np.arange(1000).reshape(200, 5)
= np.zeros(200)
y[:40] = 1
y

array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
      1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
      1., 1., 1., 1., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
      0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
      0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
      0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
      0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
      0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
      0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
      0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
      0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
      0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.])


from sklearn.model_selection import train_test_split
X_train, X_test, y_1, y_0 = train_test_split(X, y, train_size=40, shuffle = False)
#因為題目要求01個半,前面40個值都是1,所以設train_size=40抓出y_1
y_test =  y_1[:10] + y_0[:10]
print(y_0[:10])
print(y_1[:10])
print(y_test)

[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]

[1. 1. 1. 1. 1. 1. 1. 1. 1. 1.] 

[1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]

y_0, y_1都抓到,但是合併後的y_test怎麼是直接相加? 哪邊寫錯了嗎?