Is this possible?

df.sketch.ask("Which columns are integer type?")

Yes!

Sketch is an AI code-writing assistant for pandas users that understands the context of your data, greatly improving the relevance of suggestions. Sketch is usable in seconds and doesn’t require adding a plugin to your IDE.

approximatelabs/sketch: AI code-writing assistant that understands data content (github.com)