Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python - float numpy 배열에서 정수 필터링

<시간/>

데이터 정리 활동의 일부로 때때로 목록에 있는 정수를 제거해야 할 수도 있습니다. 이 기사에서 우리는 float와 integer를 모두 포함하는 배열을 가질 것입니다. 배열에서 정수를 제거하고 부동 소수점을 인쇄합니다.

astype 사용

astype 함수는 배열의 요소가 정수인지 여부를 찾는 데 사용됩니다. 따라서 배열에서 요소를 유지하거나 제거하고 결과 집합에 저장하기로 결정합니다.

import numpy as np

# initialising array
A_array = np.array([3.2, 5.5, 2.0, 4.1,5])

print("Given array :\n ", A_array)

# Only integers
res = A_array[A_array != A_array.astype(int)]

# result
print("Array without integers:\n", res)

출력

위의 코드를 실행하면 다음과 같은 결과가 나옵니다. -

Given array :
   [3.2 5.5 2. 4.1 5. ]
Array without integers:
   [3.2 5.5 4.1]

같음 및 mod 사용

이 접근 방식에서 우리는 배열의 각 요소에 mod 함수를 적용하고 나눌 때 결과가 0인지 확인합니다. 결과가 0이 아니면 부동 소수점으로 간주되어 결과로 유지됩니다.

import numpy as np

# initialising array
A_array = np.array([3.2, 5.5, 2.0, 4.1,5])

print("Given array :\n ", A_array)

# Only integers
res = A_array[~np.equal(np.mod(A_array, 1), 0)]

# result
print("Array without integers:\n", res)

출력

위의 코드를 실행하면 다음과 같은 결과가 나옵니다. -

Given array :
   [3.2 5.5 2. 4.1 5. ]
Array without integers:
   [3.2 5.5 4.1]