استخدام فهرسة المصفوفة الصحيحة لاستخراج العناصر بناءً على مؤشرات المصفوفة الثانوية

في السيناريو المحدد، الهدف هو استرداد عناصر محددة من المصفوفة A باستخدام المؤشرات المحددة في المصفوفة الثانية B. بدلاً من الاعتماد على np.take أو np.choose، هناك طريقة أكثر وضوحًا تتمثل في استخدام فهرسة مصفوفة الأعداد الصحيحة في NumPy:

A[np.arange(A.shape[0]),B.ravel()]

إليك كيفية تحقيق هذا الكود للنتيجة المرجوة:

• np.arange(A.shape[0]) ينشئ مصفوفة من الفهارس من 0 إلى (A.shape[0] - 1)، التي تمثل صفوف A.
• B.ravel() تعمل على تسطيح B في مصفوفة أحادية البعد، مما يضمن أنها تحتوي على نفس عدد العناصر كصفوف في A.
• عملية الفهرسة A[...] تسترد العناصر من A باستخدام فهارس الصفوف وفهرس الأعمدة من B.

هذا الأسلوب مفيد بشكل خاص عندما يكون B عبارة عن مصفوفة أحادية الأبعاد أو قائمة بمؤشرات الأعمدة. من خلال تخطي عملية التسطيح، يصبح الكود أكثر بساطة:

A[np.arange(A.shape[0]),B]

مثال:

A = np.array([[0, 1], [2, 3], [4, 5]])
B = np.array([1, 0, 1])

result = A[np.arange(A.shape[0]), B]

print(result)
# Output: [1, 2, 5]