numpy.sum() בפייתון

פונקציית ה-sum() של Python numpy משמשת לקבלת סכום של אלמנטים במערך מעל ציר נתון.

תחביר פונקציית ה-sum() של Python numpy

תחביר שיטת ה-sum() של Python NumPy הוא:

sum(array, axis, dtype, out, keepdims, initial)

• אלמנטים של מערך משמשים לחישוב הסכום.
• אם לא סופק ציר, מוחזר הסכום של כל האלמנטים. אם הציר הוא טאפל של מספרים שלמים, מוחזר הסכום של כל האלמנטים בצירים הנתונים.
• ניתן לציין dtype כדי לציין את סוג הנתונים של הפלט המוחזר.
• המשתנה out משמש לציון המערך להניח בו את התוצאה. זהו פרמטר אופציונלי.
• הפרמטר keepdims הוא בוליאני. אם זה מוגדר כ-True, הצירים שהופחתו נשארים בתוצאה כממדים עם גודל אחד.
• הפרמטר initial מציין את הערך ההתחלתי לסכום.

דוגמאות ל-func() sum() של Python numpy

בואו נסתכל על כמה דוגמאות לפונקציית ה-sum() של numpy.

1. סכום כל האיברים במערך

אם אנו מעבירים רק את המערך בפונקציית sum(), המערך יתוצפן ויחזיר סכום כל האיברים.

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4],
     [5, 6]])

total = np.sum(array1)
print(f'Sum of all the elements is {total}')

פלט: סכום כל האיברים הוא 21

2. סכום איברי המערך לפי ציר

אם אנו מציינים את ערך הציר, סכום האיברים לאורך הציר הזה יחזור. אם צורת המערך היא (X, Y) אז הסכום לאורך ציר 0 יהיה בצורה של (1, Y). הסכום לאורך ציר 1 יהיה בצורה של (1, X).

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4],
     [5, 6]])

total_0_axis = np.sum(array1, axis=0)
print(f'Sum of elements at 0-axis is {total_0_axis}')

total_1_axis = np.sum(array1, axis=1)
print(f'Sum of elements at 1-axis is {total_1_axis}')

פלט:

Sum of elements at 0-axis is [ 9 12]
Sum of elements at 1-axis is [ 3  7 11]

3. הגדרת סוג הנתונים המוצג של סכום

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4]])

total_1_axis = np.sum(array1, axis=1, dtype=float)
print(f'Sum of elements at 1-axis is {total_1_axis}')

פלט: סכום האיברים בציר 1 הוא [3. 7.]

4. ערך התחלתי עבור הסכום

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4]])

total_1_axis = np.sum(array1, axis=1, initial=10)
print(f'Sum of elements at 1-axis is {total_1_axis}')

פלט: סכום האיברים בציר 1 הוא [13 17] הפניה: מסמך ה- API