面试题：Python中NumPy和Pandas数据处理基础

Question

假设有一个NumPy数组 `arr = np.array([[1, 2, 3], [4, 5, 6]])`，以及一个Pandas DataFrame `df = pd.DataFrame({'A': [1, 2], 'B': [3, 4]})`。请分别使用NumPy和Pandas的方法计算数组 `arr` 中所有元素的平均值，以及DataFrame `df` 中 'A' 列的和。

Answer 1

1. 使用NumPy计算数组 arr 所有元素的平均值：

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
avg = np.mean(arr)
print(avg)

2. 使用Pandas计算DataFrame df 中 'A' 列的和：

import pandas as pd
df = pd.DataFrame({'A': [1, 2], 'B': [3, 4]})
sum_A = df['A'].sum()
print(sum_A)