用Python实现字符串分割功能

Python 笔记

一、split函数及其参数介绍

在Python中，字符串的分割可通过调用split函数来实现。该函数用于将一个字符串分割成多个子字符串，并返回一个包含所有子字符串的列表。例如：

str = "hello world"
list = str.split()
print(list) # ['hello', 'world']

split函数默认是以空格为分隔符进行分割，但是也可以自定义分隔符。

split函数还有一些可选参数，这里我们介绍其中的三个：

maxsplit：指定分割的最大次数，超过该次数后剩余部分全部作为一个元素添加到列表中。

sep：指定分割符号，默认是空格。

strip：指定是否截掉分割符号两侧的空格。

下面我们通过代码来体验这几个参数：

# maxsplit参数
str = "hello world python AI"
list = str.split(maxsplit=2)
print(list) # ['hello', 'world', 'python AI']

# sep参数
str = "hello,world,python,AI"
list = str.split(sep=",")
print(list) # ['hello', 'world', 'python', 'AI']

# strip参数
str = "  hello    world  "
list = str.split()
print(list) # ['hello', 'world']
list = str.split(maxsplit=1)
print(list) # ['hello', '   world  ']
list = str.split(maxsplit=1, strip=True)
print(list) # ['hello', 'world']

二、正则表达式进行分割

除了使用split函数外，Python还可以使用正则表达式进行分割。正则表达式是一种用来描述字符模式的方法，可用于分割字符串、匹配字符串等各种操作。

下面我们通过一个例子来学习如何使用正则表达式进行字符串分割：

import re
str = "hello,world python-mysql java;cpp"
list = re.split(r'[,;-.\s]\s*', str)
print(list) # ['hello', 'world', 'python', 'mysql', 'java', 'cpp']

正则表达式中，r表示原始字符串，‘[,;-.\s]’表示在方括号中列举所有的分割符号（包括空格、逗号等），‘\s*’表示匹配零个或多个空格，然后由re.split方法将字符串分割为多个子字符串。

三、使用特殊模块分割

除了使用split函数和正则表达式外，Python还有一些内置的模块可用于特殊的字符串分割需求。

下面我们介绍几个常用的模块：

textwrap模块提供了一些用于格式化文本字符串的函数，其中包括了一些自动分割字符串的方法。

下面我们通过一个简单的例子来演示如何使用textwrap模块分割字符串：

import textwrap
str = "hello world python AI"
list = textwrap.wrap(str, width=6)
print(list) # ['hello ', 'world ', 'python', ' AI']

textwrap.wrap函数中，width参数指定了每行字符串的最大宽度，函数将自动分割字符串并返回一个列表，其中每个元素对应一行字符串。

itertools模块是Python中的一个重要模块，提供了一组用于处理迭代器的函数。

其中，groupby函数能够将迭代器中连续的重复元素分组为单独的迭代器。

下面我们通过一个例子来演示如何使用itertools模块进行字符串分割：

import itertools
str = "helloooowooorld"
list = [''.join(group) for k, group in itertools.groupby(str)]
print(list) # ['hello', 'w', 'oo', 'world']

在该例子中，首先使用groupby函数将字符串中连续的重复字符分组为单独的迭代器，随后利用列表解析式将这些迭代器中的字符合并为单独的字符串。