如何使用 Python 从 excel 中的列中获取子字符串？

Question

我有一个 Excel 文件，我想读取该 Excel 文件中的特定列，我使用以下代码执行此操作：import pandas as pdimport xlrdfile_location = input('Where is the file located? Please input the file path here. ')column = input('In what column is the code? ')code_array = pd.read_excel(file_location, usecols=column)for i in code_array:    print(code_array)并且该代码在控制台中打印出该列的内容。现在，该列的文本如下：12345 - Description。我只想提取号码，我该怎么做？我想过使用 [0:5] 中的子字符串或将数据转换为字符串数组，但我不确定该怎么做。

Answer 1

如果数字每次都是 5 位长，您可以使用 lambda 快速创建一个子字符串。

code_array["number_column"] = code_array["YourColumnNameHere"].apply(lambda x: str(x)[:5])

如果每次的长度不一样，但是位置都一样，可以拆分成一个字符串数组，然后访问第一个元素：

code_array["number_column"] = code_array["YourColumnNameHere"].apply(lambda x: str(x).split()[0])

让我知道这是否解决了您的问题，否则我们将需要使用正则表达式。注意将 YourColumnNameHere 更改为与数据框中的列同名。