“str”对象没有属性“str”

Question

我试图在标题的任何地方打印包含tu的列。 根据String Documentation这应该可以，但是我得到一个错误

'str' object has no attribute 'str'

我试图有一个数据帧，其中的列只包含tu

import numpy as np
import pandas as pd
from sklearn.datasets import make_classification

X, y = make_classification(n_samples=1000,
                           n_features=6,
                           n_informative=3,
                           n_classes=2,
                           random_state=0,
                           scale = 10,
                           shuffle=False)

# Creating a dataFrame
df = pd.DataFrame({'car':X[:,0],
                                  'ball':X[:,1],
                                  'Feature 3': 5,
                                  'Feature 4':X[:,3],
                                  'Feature 5':X[:,4],
                                  'Feature 6':X[:,5],
                                  'Class':y})


spike_cols = [col for col in df.columns if col.str.contains("tu")]
print(spike_cols)

Answer 1

尝试使用in

spike_cols = [col for col in df.columns if "tu" in col]

看来你在找。在

^{pr2}$

Answer 2

嗯，.str不会将类型更改为字符串，请按如下方式尝试：

spike_cols = [col for col in df.columns if "tu" in str(col)]