파이썬 np.where 다중 질문

조회수 792회
import random
import numpy as np

print("몇 번 시뮬레이션 하겠습니까?>> ")
n = int(input())
 # 시뮬레이션 횟 수 입력
print("상품은 몇 개 입니까?>> ")
q = int(input())
 # 상품 수 입력
print("몇 개씩 피킹하시겠습니까?>> ")
t = int(input())
 # 피킹 토트 수 입력

A1 = (1,2,3,4,5,6,7,8,9,10,6,7,8,10,11,21,22,24,25,26,27,28,29,30,31)
arr1, arr2, arr3 = [], [], [] 
for i in range(n):                                   # 시뮬레이션 횟수 설정
    for i in range(q):                               # 상품 개수만큼 반복
        x = int(random.randrange(1, 11))             # (1~10) 사이의 난수 생성
        if x <= 7:                                   # 7 이하일시 (70%)
            ax = int(random.choice(A1))              # A1 구간에서 점수 선택
            arr1.append(ax)                          # 선택된 점수를 arr1에 정렬
        if x >7 and x<= 9 :                          # 7초과 9 이하일시 (20%)
            ax = int(random.choice(B1))              # B1 구간에서 점수 선택
            arr2.append(ax)                          # 선택된 점수를 arr2에 정렬
        if x>9 and x<=10 :                           # 9초과 10 이하일시 (10%)
            ax = int(random.choice(C1))              # C1 구간에서 점수선택
            arr3.append(ax)                   
arr1.sort(), arr2.sort(), arr3.sort()

a = np.array(arr1)
charr1 = np.where(a==22,20,a)

여기서 밑에 부분만 질문인데 arr1 안의 값이 22일때 20으로바꾸는거 까진되는데 24일땐 18, 25일땐 17 26일땐 16 이런식으로 더 조건을 추가하고 싶은데 가능할까요

  • (•́ ✖ •̀)
    알 수 없는 사용자

1 답변

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

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

(ಠ_ಠ)
(ಠ‿ಠ)