获取numpy数组中的行,其中列包含字符串

时间:2018-07-06 16:20:30

标签: python numpy

我有一个4列的numpy数组。第一列是文本。

我想检索第一列包含子字符串的数组中的每一行。

示例:如果我要搜索的字符串是“ table”,请在numpy数组中查找并返回其第一列包含“ table”的所有行。

我尝试了以下操作:

rows = nparray[searchString in nparray[:,0]]

但这似乎不起作用

1 个答案:

答案 0 :(得分:0)

给定pandas DataFrame df,它将返回所有行,其中searchString是列column中值的子字符串:

searchString = "table"

df.loc[df['column'].str.contains(searchString, regex=False)]