파이썬 행렬 질문: 오른쪽 맨위, 왼쪽 맨위, 오른쪽 맨 아래, 왼쪽 맨 아래 만 1이고 나머지는 0 으로 만드는거 어떻게 하나요?

조회수 738회

오른쪽 맨위, 왼쪽 맨위, 오른쪽 맨 아래, 왼쪽 맨 아래 만 1이고 나머지는 0 으로 만드는거 어떻게 하나요?

[[1 0 0],
[0 0 0],
[0 0 0]]

[[ 0 0 1],
[0 0 0],
[0 0 0]]

[[0 0 0],
[0 0 0],
[0 0 1]]

[[ 0 0],
[0 0 0],
[1 0 0]]

1 답변

  • >>> import numpy as np
    >>> a = np.zeros((3, 3))
    >>> a
    array([[0., 0., 0.],
           [0., 0., 0.],
           [0., 0., 0.]])
    >>> a[0,0] = 1
    >>> a
    array([[1., 0., 0.],
           [0., 0., 0.],
           [0., 0., 0.]])
    >>> a[-1, 0] = 1
    >>> a
    array([[1., 0., 0.],
           [0., 0., 0.],
           [1., 0., 0.]])
    >>> a[0, -1] = 1
    >>> a
    array([[1., 0., 1.],
           [0., 0., 0.],
           [1., 0., 0.]])
    >>> a[-1, -1] = 1
    >>> a
    array([[1., 0., 1.],
           [0., 0., 0.],
           [1., 0., 1.]])
    

답변을 하려면 로그인이 필요합니다.

프로그래머스 커뮤니티는 개발자들을 위한 Q&A 서비스입니다. 로그인해야 답변을 작성하실 수 있습니다.

(ಠ_ಠ)
(ಠ‿ಠ)