Как получить возвращаемое значение количества уникальных значений в NumPy?

Question

У меня есть массив «станция», который содержит два значения [28450., 29980.]. Они представляют названия станций. Я хочу получить количество этих станций в новой переменной, используя numpy:

station_count = np.unique(station)
print("The number of unique stations is ", station_count)

дает мне:

Количество уникальных станций составляет [28450. 29980.]

Как мне получить номер 2 вместо [28450. 29980.]. Я пробовал считать, но мне кажется, что синтаксис не корректен.

Answer 1

Вы можете использовать len уникального массива:

station_count = len(np.unique(station))
print("The number of unique stations is ", station_count)

Answer 2

Вы можете получить больше информации по коду, указанному ниже: -

import numpy as np

x = np.array([1,1,1,2,2,2,5,25,1,1])
unique, counts = np.unique(x, return_counts=True)
print(f"The number of unique values are \t {len(unique)}")
print()
for value, number in zip(unique,counts):
    print(f"The value {value} repeats \t -->> {number} times")

Надеюсь, вы узнаете что-то новое, связанное с вашим вопросом.