Python代码优化之如何使用更高效的数据结构

代码优化是提高Python程序性能的有效方式。下面web建站小编给大家简单介绍一下如何使用更高效的数据结构！

列表查询：用集合代替列表,可以提高查询效率。字典查询：​使用defaultdict可以避免KeyError。用pandas DataFrame和Series替换纯列表。

示例代码如下：

import collections
import pandas as pd
list1 = [1, 2, 6, 4, 2, 3, 2]

# 普通列表查询
list1.count(2)

# 使用集合
set1 = set(list1)
list1.count(2)

# 使用defaultdict避免KeyError
dd = collections.defaultdict(int)
for k in list1:
dd[k] += 1
dd[2]

# 使用pandas
df = pd.DataFrame({'col1': [1, 2, 6, 4, 2, 3, 2]})
df['col1'].value_counts()[2]