时间戳

首页 > 解决方案 > 从熊猫数据框中提取单个值

python - 从熊猫数据框中提取单个值

问题描述

在 Python 中，我试图从 Pandas 数据框中提取单个值。我确切地知道该值包含什么，我只需要在数据框中的任何位置找到它并提取它。

例如，在下面的数据框中：

df = pd.DataFrame(
        {0: ['BA1234', 'CA:1234', 'DA','DA1234', 'EX DA', 'CA1234'],
         1: ['BA1234', 'CA:1234', 'DA','CA1234', 'EX DA', 'CA1234'],
         2: ['BA1234', 'CA:1234', 'DA','CA1234', 'EX DA', 'CA1234']})

我想提取包含两个字母 'DA' 和后面正好 4 个数字的字符串。

我一直在用面具尝试这个：

mask = pd.DataFrame(np.column_stack([df[col].str.contains('^DA\d{4}', na = False) for col in df]))

这似乎有效：

da_value = df[mask]

da_value
        0    1    2
0     NaN  NaN  NaN
1     NaN  NaN  NaN
2     NaN  NaN  NaN
3  DA1234  NaN  NaN
4     NaN  NaN  NaN
5     NaN  NaN  NaN

但是，如何从数据框中提取值？有没有更好/更简单的方法来做到这一点？

编辑：我真正想要的输出是

da_value = 'DA1234'

标签： pythonpandasdataframe

解决方案

先用DataFrame.stackfor ，然后用withSeries过滤：boolean indexingSeries.str.contains

s = df.stack()
a = s[s.str.contains(r'^DA\d{4}', na=False)].tolist()

如果需要列表中的第一个值，您可以选择：

print (a[0])
DA1234

或通用解决方案，如果可能不存在值，则添加默认值：

print (next(iter(a), 'no match'))
DA1234

推荐阅读