Python基础——使用with结构打开多个文件

考虑如下的案例:
同时打开三个文件,文件行数一样,要求实现每个文件依次读取一行,然后输出,我们先来看比较容易想到的写法:

with open(filename1, 'rb') as fp1:
    with open(filename2, 'rb') as fp2:
        with open(filename3, 'rb') as fp3:
            for i in fp1:
                j = fp2.readline()
                k = fp3.readline()
                print(i, j, k)

注意这里只能对单个文件进行for循环读取,不能写成:

for i, j, k in fp1, fp2, fp3:
    print(i, j, k)

但可使用强大的zip操作:

for i, j, k in zip(fp1, fp2, fp3):
    print(i, j, k)

这样层层的嵌套未免啰嗦,with结构支持一种更简洁的写法:

with open(filename1, 'rb') as fp1, open(filename2, 'rb') as fp2, open(filename3, 'rb') as fp3:
    for i in fp1:
        j = fp2.readline()
        k = fp3.readline()
        print(i, j, k)

或者使用更为优雅的写法,此时需要contextlib语法糖:

from contextlib improt ExitStack
with ExitStack() as stack:
    files = [stack.enter_context(open(fname)) for fname in [filename1, filename2, filename3]]
    for i, j, k in zip(files[0], files[1], files[2]):
        print(i, j, k)
  • 5
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
好的,我了解了你的问题。你想要编写一个 ArcGIS Python 脚本,能够根据要素类(shp)的一个字段内容,将这个要素类拆分成多个要素类,对吗? 那么,你可以按照以下步骤来编写这个脚本: 1. 首先,导入必要的 Python 模块和 ArcGIS 模块,包括 arcpy 和 os 模块。 ``` import arcpy import os ``` 2. 定义输入要素类的路径和字段名称。 ``` in_fc = r"path/to/input/feature/class.shp" split_field = "field_name" ``` 3. 使用 arcpy.da.SearchCursor 函数遍历要素类中的每个要素,并根据指定字段的值创建一个字典,其中键为字段值,值为要素类中包含该字段值的所有要素的列表。 ``` split_dict = {} with arcpy.da.SearchCursor(in_fc, [split_field]) as cursor: for row in cursor: if row[0] not in split_dict: split_dict[row[0]] = [] split_dict[row[0]].append(row) ``` 4. 遍历字典中的每个键值对,使用 arcpy.CopyFeatures_management 函数将包含每个键值对所对应字段值的所有要素复制到一个新的要素类中。 ``` out_folder = r"path/to/output/folder" for value, rows in split_dict.items(): out_name = "split_" + str(value) + ".shp" out_path = os.path.join(out_folder, out_name) arcpy.CopyFeatures_management(rows, out_path) ``` 5. 完整的代码如下所示: ``` import arcpy import os in_fc = r"path/to/input/feature/class.shp" split_field = "field_name" out_folder = r"path/to/output/folder" split_dict = {} with arcpy.da.SearchCursor(in_fc, [split_field]) as cursor: for row in cursor: if row[0] not in split_dict: split_dict[row[0]] = [] split_dict[row[0]].append(row) for value, rows in split_dict.items(): out_name = "split_" + str(value) + ".shp" out_path = os.path.join(out_folder, out_name) arcpy.CopyFeatures_management(rows, out_path) ``` 这个脚本将会根据输入要素类的指定字段值,将要素类拆分成多个要素类,并将这些要素类保存到指定的输出文件夹中。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

五道口纳什

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值