본문 바로가기
파이썬

파이썬 Assigning to columns in NumPy?

by º기록 2021. 2. 15.
반응형

NumPy를 사용하여 다음 MATLAB 코드를 어떻게 작성할 수 있습니까?

A = zeros(5, 100);
x = ones(5,1);
A(:,1) = x;

행에 할당하는 것은 쉽게 작동하는 것 같지만 다른 배열의 열에 배열을 할당하는 예를 찾을 수 없습니다.

 

해결 방법

 

a [:, 1] = x [:, 0] 을 사용합니다. x 의 열을 단일 numpy 배열로 선택하려면 x [:, 0] 이 필요합니다. x 의 형식을 지정하는 방법을 선택할 수있는 경우 처음에는 2 차원 배열이 아닌 일반 (행) 배열로 만드는 것이 좋습니다.

>>> a
array([[ 0.,  0.,  0.],
       [ 0.,  0.,  0.],
       [ 0.,  0.,  0.],
       [ 0.,  0.,  0.],
       [ 0.,  0.,  0.]])
>>> x = numpy.ones(5)
>>> x
array([ 1.,  1.,  1.,  1.,  1.])
>>> a[:,1] = x
>>> a
array([[ 0.,  1.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  1.,  0.]])

 

참조 페이지 https://stackoverflow.com/questions/10923212

 

 

반응형

댓글