Как эффективно разбросать участок массива numpy 2d

Question

У меня есть numpy с каждой строкой, содержащей пары x, y, и я хочу отобразить точечный график без использования for для l oop, поэтому я использовал следующий подход, используя pandas:

def visualize_k_means_output(self, centroids):
    fig, ax = plt.subplots()
    self.visualize_box_relative_sizes()
    frame = pd.DataFrame(centroids, columns=['X', 'Y'])
    ax.scatter(frame['X'], frame['Y'], marker='*', s=200, c='black')

Вопрос в том, как извлечь первый элемент как x, а второй элемент как y, не используя a для l oop, например:

ax.scatter(x=[item[0] for item in centroids], y=[item[1] for item in centroids], ...)

Answer 1

Если я правильно понял, вы хотите нарезать массив numpy:

x = centroids[:, 0]
y = centroids[:, 1]