Payload: Pandas and Numpy
Easy
1. Create a NumPy Array
Write a Python program to create a NumPy array with values ranging from 1 to 10.
o Input: None
o Output:
o array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
2. Select a Column from a DataFrame
Given the DataFrame below, write a Pandas program to select the Name column.
o Input:
Python
df = pd.DataFrame({
'Name': ['Alice', 'Bob', 'Charlie'],
'Age': [25, 30, 35],
'City': ['New York', 'Los Angeles', 'Chicago']
})
o Output:
o 0 Alice
o 1 Bob
o Name: Name, dtype: object
3. Create a DataFrame from a Dictionary
Write a Python program to create a Pandas DataFrame from the dictionary below.
o Input:
Python
data = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}
o Output:
o A BC
o 0 1 4 7
o 1 2 5 8
o 2 3 6 9
Medium
4. Find the Mean of a NumPy Array
Write a Python program to find the mean of a given NumPy array.
o Input:
o array([10, 20, 30, 40, 50])
o Output:
o 30.0
5. Filter Rows by Condition in Pandas
Given the DataFrame below, write a Pandas program to filter rows where the Age is
greater than 30.
o Input:
Python
df = pd.DataFrame({
'Name': ['Alice', 'Bob', 'Charlie'],
'Age': [25, 30, 35],
'City': ['New York', 'Los Angeles', 'Chicago']
})
o Output:
o Name Age City
o 2 Charlie 35 Chicago
6. Generate a Random NumPy Array
Write a Python program to generate a 3x3 matrix of random values between 0 and 1
using NumPy.
o Input: None
o Output: (Values may vary)
o array([[0.329, 0.957, 0.604],
o [0.158, 0.940, 0.136],
o [0.410, 0.234, 0.732]])
7. Calculate the Sum of Each Column in a DataFrame
Given the DataFrame below, write a Pandas program to calculate the sum of each
column.
o Input:
Python
df = pd.DataFrame({
'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]
})
o Output:
o A 6
o B 15
o C 24
o dtype: int64
8. Replace Missing Values in a DataFrame
Write a Pandas program to replace missing values (NaN) in the DataFrame with the
mean of the respective column.
o Input:
Python
df = pd.DataFrame({
'A': [1, np.nan, 3],
'B': [4, 5, np.nan],
'C': [7, 8, 9]
})
o Output:
o A B C
o 0 1.0 4.0 7
o 1 2.0 5.0 8
o 2 3.0 4.5 9
Hard
9. Find the Dot Product of Two NumPy Arrays
Write a Python program to calculate the dot product of two NumPy arrays.
o Input:
Python
array_1 = np.array([1, 2, 3])
array_2 = np.array([4, 5, 6])
o Output:
o 32
10. Group By and Aggregate in Pandas
Given the DataFrame below, write a Pandas program to group by the City column
and calculate the mean Age for each city.
• Input:
Python
df = pd.DataFrame({
'Name': ['Alice', 'Bob', 'Charlie', 'David'],
'Age': [25, 30, 35, 40],
'City': ['New York', 'Los Angeles', 'Chicago', 'New York']
})
• Output:
• Age
• City
• Chicago 35.0
• Los Angeles 30.0
• New York 32.5
11. Apply a Function to a DataFrame Column
Write a Python program to apply a custom function to the Age column of the
DataFrame below to convert the ages from years to months.
• Input:
Python
df = pd.DataFrame({
'Name': ['Alice', 'Bob', 'Charlie'],
'Age': [25, 30, 35]
})
def years_to_months(age):
return age * 12
• Output:
• Name Age
• 0 Alice 300
• 1 Bob 360
• 2 Charlie 420
12. Concatenate DataFrames
Write a Pandas program to concatenate the two DataFrames below.
• Input:
Python
df1 = pd.DataFrame({'A': [1, 2], 'B': [3, 4]})
df2 = pd.DataFrame({'A': [5, 6], 'B': [7, 8]})
• Output:
• AB
• 0 1 3
• 1 2 4
• 0 5 7
• 1 6 8
13. Find the Eigenvalues and Eigenvectors
Write a Python program to find the eigenvalues and eigenvectors of a 2x2 matrix
using NumPy.
• Input:
Python
matrix = np.array([[4, 2], [1, 3]])
• Output:
• Eigenvalues: [5.236, 1.764]
• Eigenvectors:
• [[ 0.894, -0.707],
• [ 0.447, 0.707]]
14. Reshape a NumPy Array
Write a Python program to reshape a 1D NumPy array into a 3x3 matrix.
• Input:
Python
array = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
• Output:
• array([[1, 2, 3],
• [4, 5, 6],
• [7, 8, 9]])
15. Pivot a DataFrame
Given the DataFrame below, write a Pandas program to pivot the DataFrame so that
the Name column becomes the index, and the Subject column values become
columns with the corresponding Score.
• Input:
Python
df = pd.DataFrame({
'Name': ['Alice', 'Bob', 'Alice', 'Bob'],
'Subject': ['Math', 'Math', 'Science', 'Science'],
'Score': [85, 78, 92, 81]
})
• Output:
• Math Science
• Name
• Alice 85 92
• Bob 78 81
This sheet covers fundamental and advanced topics in Pandas and NumPy,
enabling you to test a broad range of data manipulation and computational skills.
