Meaning of X = X[:, 1] in Python

后端 未结 3 452
长情又很酷
长情又很酷 2021-01-31 09:41

I am studying this snippet of python code. What does X = X[:, 1] mean in last line?

def linreg(X,Y):
    # Running the linear regression
    X = sm.         


        
3条回答
  •  面向向阳花
    2021-01-31 09:45

    x = np.random.rand(3,2)
    
    x
    Out[37]: 
    array([[ 0.03196827,  0.50048646],
           [ 0.85928802,  0.50081615],
           [ 0.11140678,  0.88828011]])
    
    x = x[:,1]
    
    x
    Out[39]: array([ 0.50048646,  0.50081615,  0.88828011])
    

    So what that line did is sliced the array, taking all rows (:) but keeping the second column (1)

提交回复
热议问题