Question

我正在尝试在标题中的任何位置打印包含tu的列。根据{{3}}，这应该有效，但我收到错误

'str'对象没有属性'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

嗯，.str没有将类型更改为字符串，请尝试这样：

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

Answer 2

尝试使用in

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

看起来你正在寻找。

df.filter(like='tu')

'str'对象没有属性'str'

2 个答案: