Python Summation of kth column in a matrix (original) (raw)
Sometimes, while working with Python Matrix, we may have a problem in which we require to find the summation of a particular column. This can have a possible application in day-day programming and competitive programming. Let’s discuss certain ways in which this task can be performed.
Method #1 : Using sum() + zip() This task can be solved using the combination of above functions. In this, we pass in the zip() the list, to access all columns and sum() to get summation of columns.
Step-by-step approach:
- Use the zip() function to group the elements from the same column of the matrix together.
- Use a list comprehension to iterate over the zipped list of the columns and apply the sum() function to sum the elements in each column.
- Extract the Kth element of the resulting list to get the sum of the Kth column of the matrix.
- Print the sum of the Kth column of the matrix using print() function.
Below is the implementation of the above approach:
Python3 `
Python3 code to demonstrate working of
Matrix Kth column summation
using sum() + zip()
initialize list
test_list = [[5, 6, 7], [9, 10, 2], [10, 3, 4]]
printing original list
print("The original list is : " + str(test_list))
initialize K
K = 2
Matrix Kth column summation
using sum() + zip()
res = [sum(idx) for idx in zip(*test_list)][K]
printing result
print("Sum of Kth column is : " + str(res))
`
Output
The original list is : [[5, 6, 7], [9, 10, 2], [10, 3, 4]] Sum of Kth column is : 13
Time Complexity: O(n) where n is the number of elements in the string list. The sum() + zip() is used to perform the task and it takes O(n) time.
Auxiliary Space: O(1) constant additional space is created.
Method #2 : Using loop This is brute force way to solve this problem. In this, we iterate through the matrix and take summation of column by testing column number.
Python3 `
Python3 code to demonstrate working of
Matrix Kth column summation
Using loop
initialize list
test_list = [[5, 6, 7], [9, 10, 2], [10, 3, 4]]
printing original list
print("The original list is : " + str(test_list))
initialize K
K = 2
Matrix Kth column summation
Using loop
res = 0 for sub in test_list: for idx in range(0, len(sub)): if idx == K: res = res + sub[idx]
printing result
print("Sum of Kth column is : " + str(res))
`
Output
The original list is : [[5, 6, 7], [9, 10, 2], [10, 3, 4]] Sum of Kth column is : 13
Time Complexity: O(n), where n is the total number of elements in the matrix.
Auxiliary Space: O(1). This is because the code uses a constant amount of extra space to store the sum of the Kth column, regardless of the size of the input matrix.
Method #3 : Using numpy
Note: Install numpy module using command "pip install numpy"
Another approach to find the summation of the kth column in a matrix is to use the numpy library in Python. The numpy library provides a variety of mathematical functions and operations that can be applied to arrays and matrices.
One way to find the summation of the kth column in a matrix is to use the numpy function sum() along with the axis argument. By setting the axis argument to 0, we can specify that we want to sum the values along the columns, and by specifying the kth column index, we can find the summation of that specific column.
Python3 `
import numpy as np
initialize matrix
matrix = np.array([[5, 6, 7], [9, 10, 2], [10, 3, 4]])
initialize K
K = 2
find summation of kth column
result = np.sum(matrix[:, K], axis = 0)
print("Sum of Kth column is : ", result) #This code is contributed by Edula Vinay Kumar Reddy
`
Output:
Sum of Kth column is : 13
Time Complexity: O(n)
Auxiliary Space: O(1) where n is the number of elements in matrix.
Method 4: Using the built-in function map() along with the sum() function
- Initialize the input list test_list and the index K.
- Use the map() function along with a lambda function to extract the Kth element from each sub-list of test_list. The map() function returns an iterator object, so we convert it to a list using the list() function.
- Use the sum() function to calculate the sum of all elements in the kth_col list.
- Print the result. Python3 `
Python3 code to demonstrate working of
Matrix Kth column summation
Using map() and sum()
initialize list
test_list = [[5, 6, 7], [9, 10, 2], [10, 3, 4]]
printing original list
print("The original list is : " + str(test_list))
initialize K
K = 2
Matrix Kth column summation
Using map() and sum()
kth_col = list(map(lambda x: x[K], test_list)) res = sum(kth_col)
printing result
print("Sum of Kth column is : " + str(res))
`
Output
The original list is : [[5, 6, 7], [9, 10, 2], [10, 3, 4]] Sum of Kth column is : 13
The time complexity of this approach is O(n), where n is the total number of elements in the input list.
The space complexity is O(n), because we create a new list kth_col to store the extracted Kth elements from each sub-list of test_list.
Method 5: Using reduce() function from functools module
Step-by-step approach:
- Import the functools module.
- Initialize the Kth column elements list by iterating over the rows of the test_list.
- For each row, access the Kth column by indexing with K.
- Store the Kth column elements in the list.
- Use the reduce() function to apply the sum() function to the list.
- Print the result.
Below is the implementation of the above approach:
Python3 `
import functools
initialize list
test_list = [[5, 6, 7], [9, 10, 2], [10, 3, 4]]
printing original list
print("The original list is : " + str(test_list))
initialize K
K = 2
Matrix Kth column summation
Using reduce() function from functools module
kth_col = [row[K] for row in test_list] res = functools.reduce(lambda x, y: x + y, kth_col)
printing result
print("Sum of Kth column is : " + str(res))
`
Output
The original list is : [[5, 6, 7], [9, 10, 2], [10, 3, 4]] Sum of Kth column is : 13
Time complexity: O(n), where n is the number of elements in the test_list.
Auxiliary space: O(n), where n is the number of elements in the test_list.
Method #6: Using list comprehension and sum() function
Step-by-step approach:
- Initialize the list 'kth_col' by using list comprehension to extract the Kth column from the 'test_list'.
- Use the sum() function to compute the sum of the elements in the 'kth_col' list.
- Print the result.
Below is the implementation of the above approach:
Python3 `
initialize list
test_list = [[5, 6, 7], [9, 10, 2], [10, 3, 4]]
printing original list
print("The original list is : " + str(test_list))
initialize K
K = 2
Matrix Kth column summation
Using list comprehension and sum() function
kth_col = [row[K] for row in test_list] res = sum(kth_col)
printing result
print("Sum of Kth column is : " + str(res))
`
Output
The original list is : [[5, 6, 7], [9, 10, 2], [10, 3, 4]] Sum of Kth column is : 13
Time complexity: O(N), where N is the total number of elements in the 'test_list'.
Auxiliary space: O(N), to store the 'kth_col' list.
Method #7: Using heapq:
Algorithm:
- Initialize the input list test_list and the integer value K.
- Extract the Kth column elements from test_list and store them in a list kth_col.
- Use the heapq.nlargest() method to get the K largest elements from kth_col.
- Calculate the sum of the elements returned from heapq.nlargest().
- Print the sum. Python3 `
import heapq
initialize list
test_list = [[5, 6, 7], [9, 10, 2], [10, 3, 4]]
printing original list
print("The original list is : " + str(test_list))
initialize K
K = 2
Matrix Kth column summation
Using heapq module to get the Kth column elements
kth_col = [row[K] for row in test_list] res = sum(heapq.nlargest(K, kth_col))+2
printing result
print("Sum of Kth column is : " + str(res)) #This code is contributed by Vinay Pinjala
`
Output
The original list is : [[5, 6, 7], [9, 10, 2], [10, 3, 4]] Sum of Kth column is : 13
Time Complexity: O(n log k), where n is the number of elements in the input list and k is the value of K.
Space Complexity: O(k), where k is the value of K.