如何优化Python中的算法和数据结构

# 普通二维数组相加
def matrix_sum(matrix):
    result = 0
    for i in range(len(matrix)):
        for j in range(len(matrix[i])):
            result += matrix[i][j]
    return result

# 使用迭代器替代循环嵌套
def matrix_sum(matrix):
    result = 0
    for row in matrix:
        for element in row:
            result += element
    return result

# 使用循环生成平方数列表
def square_numbers(n):
    result = []
    for i in range(1, n+1):
        result.append(i**2)
    return result

# 使用列表生成式生成平方数列表
def square_numbers(n):
    return [i**2 for i in range(1, n+1)]

# 使用列表和循环查找重复元素
def find_duplicates(numbers):
    duplicates = []
    for i in range(len(numbers)):
        if numbers.count(numbers[i]) > 1:
            if numbers[i] not in duplicates:
                duplicates.append(numbers[i])
    return duplicates

# 使用集合和列表生成式查找重复元素
def find_duplicates(numbers):
    return [number for number in set(numbers) if numbers.count(number) > 1]

# 使用自定义字典统计单词频率
def Word_frequency(text):
    word_dict = {}
    for word in text.split():
        if word not in word_dict:
            word_dict[word] = 1
        else:
            word_dict[word] += 1
    return word_dict

# 使用内置字典统计单词频率
def word_frequency(text):
    word_dict = {}
    for word in text.split():
        word_dict[word] = word_dict.get(word, 0) + 1
    return word_dict

# 使用内置列表查找最大值
def find_max(numbers):
    max_number = numbers[0]
    for number in numbers:
        if number > max_number:
            max_number = number
    return max_number

# 使用内置堆查找最大值
import heapq
def find_max(numbers):
    return heapq.nlargest(1, numbers)[0]