기타

파이썬 배열 1보다 큰 행과 열을 찾는 코드

철인31호 2024. 5. 17. 10:45
반응형

파이썬 배열에서 모든 요소를 검색하여 1보다 큰 행과 열을 찾는 코드를 작성하는 방법을 설명하겠습니다. 이 코드는 2차원 배열(리스트)을 입력으로 받아서, 각 요소를 검사한 후, 1보다 큰 요소가 있는 행과 열을 출력합니다. 아래에 코드와 함께 블로그 포스팅 형식으로 작성해 보았습니다.

---

### 파이썬 배열에서 1보다 큰 요소를 찾기

파이썬을 사용하여 2차원 배열에서 모든 요소를 검색하고, 1보다 큰 값을 포함하는 행과 열을 찾는 방법에 대해 알아보겠습니다. 이 과정은 간단하지만, 배열 탐색 및 조건 검사에 대한 기본적인 이해를 필요로 합니다.

#### 예제 코드

우선, 2차원 배열을 생성하고, 각 요소를 검색하여 조건을 만족하는 행과 열을 출력하는 코드를 작성해 보겠습니다.

```python
# 2차원 배열 예제
array = [
    [0, 2, 0],
    [1, 1, 3],
    [0, 0, 0]
]

# 1보다 큰 값을 포함하는 행과 열을 찾는 함수
def find_rows_and_columns_larger_than_one(array):
    rows = set()
    cols = set()

    # 배열의 모든 요소를 검사
    for i in range(len(array)):
        for j in range(len(array[i])):
            if array[i][j] > 1:
                rows.add(i)
                cols.add(j)

    return list(rows), list(cols)

# 함수 호출 및 결과 출력
rows, cols = find_rows_and_columns_larger_than_one(array)
print("1보다 큰 값을 포함하는 행: ", rows)
print("1보다 큰 값을 포함하는 열: ", cols)
```

#### 코드 설명

1. **배열 초기화**: `array` 변수에 2차원 배열을 정의합니다.
2. **함수 정의**: `find_rows_and_columns_larger_than_one` 함수를 정의합니다. 이 함수는 배열을 입력받아 1보다 큰 값을 포함하는 행과 열을 찾습니다.
3. **행과 열 저장**: 행과 열을 저장할 `rows`와 `cols` 집합을 생성합니다. 집합(set)을 사용하면 중복을 자동으로 제거할 수 있습니다.
4. **배열 탐색**: 이중 for 루프를 사용하여 배열의 모든 요소를 검사합니다. 만약 요소가 1보다 크다면, 해당 요소의 행 인덱스와 열 인덱스를 각각 `rows`와 `cols` 집합에 추가합니다.
5. **결과 반환**: 집합을 리스트로 변환하여 반환합니다.
6. **결과 출력**: 함수를 호출하여 결과를 출력합니다.

#### 결과 해석

위 코드의 배열에서 1보다 큰 값은 2와 3입니다. 2는 첫 번째 행의 두 번째 열에 있으며, 3은 두 번째 행의 세 번째 열에 있습니다. 따라서 함수는 첫 번째와 두 번째 행, 그리고 두 번째와 세 번째 열을 출력합니다.

- **출력 결과**:
  ```
  1보다 큰 값을 포함하는 행:  [0, 1]
  1보다 큰 값을 포함하는 열:  [1, 2]
  ```

#### 결론

이 코드는 2차원 배열에서 조건을 만족하는 행과 열을 찾는 데 유용합니다. 이를 통해 배열의 특정 값을 검색하고, 그 위치를 쉽게 식별할 수 있습니다. 이러한 기술은 데이터 분석, 이미지 처리, 행렬 연산 등 다양한 분야에서 활용될 수 있습니다.

---

이렇게 배열 탐색 및 조건 검사를 활용하여 원하는 정보를 추출하는 방법을 알아보았습니다. 이 코드를 통해 여러분의 파이썬 프로그래밍 실력을 더욱 향상시키길 바랍니다.

반응형