PyMongo导入CSV：数值转换技巧分享

积累知识，胜过积蓄金银！毕竟在文章开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《PyMongo导入CSV：数值类型转换教程》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

本教程详细介绍了如何使用PyMongo将CSV文件导入MongoDB时，解决所有字段被默认为字符串类型的问题。核心方法是利用Python的`csv.DictReader`读取数据后，在插入数据库前对特定的数值字段（如整数和浮点数）进行显式类型转换。通过示例代码，您将学习如何高效地处理数据类型，确保数据以正确的格式存储在MongoDB中，从而避免后续的数据处理困扰。

引言：CSV数据导入MongoDB的常见挑战

在使用Python的PyMongo库将CSV（Comma Separated Values）文件导入MongoDB时，一个常见的挑战是数据类型处理。由于CSV文件本质上是纯文本格式，Python的csv模块（特别是csv.DictReader）在读取时会将所有字段的内容都视为字符串。这意味着，即使CSV文件中包含数字（如整数或浮点数），在未经额外处理的情况下，它们也会作为字符串类型存储到MongoDB中。这可能导致后续的查询、聚合或数值计算出现问题。

问题分析：为什么数据会以字符串形式存储？

当我们使用csv.DictReader读取CSV文件时，它会迭代每一行，并将每行数据解析为一个字典，其中键是CSV文件的列头，值是对应的单元格内容。然而，csv.DictReader本身并没有内置的类型推断或转换机制，它只是简单地将读取到的文本内容作为字符串返回。

考虑以下原始的Python代码片段，它展示了这种默认行为：

import csv
from pymongo import MongoClient

def saveToMongo():
    print("inside saveToMongo")
    # 连接MongoDB（此处省略了连接细节，假设mydb已定义）
    # collection = mydb['country'] 

    header = ['country_id','country_name','zone_id','minLat','maxLat','minLong','maxLong']
    csvFile = open('country.csv', 'r')
    reader = csv.DictReader(csvFile)

    for each in reader:
        row = {}
        for field in header:
            row[field] = each[field] # 这里each[field]已经是字符串
        # collection.insert(row) # 插入时，所有字段仍是字符串
    csvFile.close()

# 示例CSV文件内容：
# country_id,country_name,zone_id,minLat,maxLat,minLong,maxLong
# 2,Bangladesh,1,20.6708832870000,26.4465255803000,88.0844222351000,92.6727209818000

在这段代码中，each[field]从csv.DictReader中获取的值始终是字符串。因此，即使像country_id或minLat这样的字段在CSV中看起来是数字，它们也会以字符串形式被赋值给row[field]，并最终以字符串类型插入到MongoDB中。

解决方案：在插入前进行显式类型转换

解决这个问题的关键是在数据被插入MongoDB之前，对需要数值类型的字段进行显式的数据类型转换。Python提供了int()和float()等内置函数，可以将字符串转换为对应的整数或浮点数。

以下是优化后的Python代码，它演示了如何实现这一转换：

from pymongo import MongoClient 
import csv

# 1. 连接MongoDB数据库
# 假设MongoDB运行在本地默认端口27017
myclient = MongoClient("mongodb://localhost:27017/")
mydb = myclient["mydbname"] # 替换为你的数据库名称

def csvToMongo(csv_file_path='country.csv', collection_name='country'):
  """
  从CSV文件读取数据，进行类型转换后批量插入MongoDB。

  Args:
    csv_file_path (str): CSV文件的路径。
    collection_name (str): 要插入数据的MongoDB集合名称。
  """
  collection = mydb[collection_name]

  # 使用with语句确保文件被正确关闭
  with open(csv_file_path, 'r', encoding='utf-8') as myFile:
    reader = csv.DictReader(myFile, delimiter=",")

    myParsedData = []
    for elem in reader:
      # 对每个字段进行显式类型转换
      parsed_row = {
        'country_id': int(elem['country_id']),
        'country_name': elem['country_name'],
        'zone_id': int(elem['zone_id']),
        'minLat': float(elem['minLat']),
        'maxLat': float(elem['maxLat']),
        'minLong': float(elem['minLong']),
        'maxLong': float(elem['maxLong']),
      }
      myParsedData.append(parsed_row)

    # 使用insert_many进行批量插入，提高效率
    if myParsedData: # 确保有数据才进行插入
        collection.insert_many(myParsedData)
        print(f"成功将 {len(myParsedData)} 条数据插入到集合 '{collection_name}'。")
    else:
        print("CSV文件为空或没有可解析的数据。")

# 执行函数
if __name__ == "__main__":
    csvToMongo()

代码解析：

1. MongoDB连接: myclient = MongoClient("mongodb://localhost:27017/") 建立了与MongoDB服务器的连接。mydb = myclient["mydbname"] 选择了要操作的数据库。
2. 文件读取: with open('country.csv','r') as myFile: 以只读模式打开CSV文件。with语句确保文件在操作完成后自动关闭，即使发生错误。encoding='utf-8' 是推荐的文件编码设置，以避免字符编码问题。
3. csv.DictReader: reader = csv.DictReader(myFile, delimiter=",") 创建一个字典阅读器对象，它将CSV的每一行解析为字典。delimiter="," 明确指定了分隔符。
4. 数据类型转换: 这是核心步骤。通过一个循环或列表推导式，我们遍历reader中的每一行数据（elem），并对需要转换为数值类型的字段进行显式转换：
   • int(elem['country_id']): 将country_id字段的字符串值转换为整数。
   • float(elem['minLat']): 将minLat字段的字符串值转换为浮点数（Python中的float对应MongoDB的double）。
   • 对于不需要转换的字段（如country_name），直接保留其字符串格式。
5. 批量插入: collection.insert_many(myParsedData) 是PyMongo推荐的批量插入方式。它比在循环中多次调用insert_one（或已弃用的insert）效率更高，因为它只需要一次网络往返就能插入多条文档。

注意事项与最佳实践

• 错误处理: 在实际应用中，CSV文件中的数据可能不总是干净的。例如，如果minLat字段包含非数字字符，float()转换会抛出ValueError。为了提高代码的健壮性，您可以使用try-except块来捕获这些错误，并处理异常数据（例如，跳过该行、记录错误或赋默认值）。

  try:
      parsed_row['minLat'] = float(elem['minLat'])
  except ValueError:
      print(f"警告: 无法将 '{elem['minLat']}' 转换为浮点数，跳过此行或设置为默认值。")
      # 可以选择跳过此行 continue
      # 或者设置为None parsed_row['minLat'] = None
      # 或者设置为默认值 parsed_row['minLat'] = 0.0

• 文件编码: 始终指定CSV文件的正确编码（如encoding='utf-8'），以避免中文或其他特殊字符乱码问题。

• 字段映射: 如果CSV文件的列名与您希望在MongoDB中使用的字段名不同，可以在构建parsed_row字典时进行映射。

• mongoimport工具: 虽然本教程侧重于PyMongo的解决方案，但MongoDB官方也提供了mongoimport命令行工具，它在导入CSV时支持通过--type=csv和--columnsHaveTypes等选项进行一定程度的类型推断或指定类型。然而，对于复杂的转换逻辑或需要与Python应用深度集成的情况，PyMongo提供了更大的灵活性。

• 性能优化: 对于非常大的CSV文件，可以考虑分批读取和插入，而不是一次性将所有数据加载到内存中。

总结

通过在PyMongo导入CSV数据时，对读取到的字符串字段进行显式的数据类型转换，我们可以确保数据以正确的格式存储在MongoDB中。这种方法提供了精确的控制，并且通过insert_many可以高效地完成批量数据导入。在实际项目中，结合错误处理机制，能够构建出更加健壮和可靠的数据导入流程。

本篇关于《PyMongo导入CSV：数值转换技巧分享》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于文章的相关知识，请关注golang学习网公众号！