To read a specific column from a CSV file, you should use the usecols parameter in pandas' read_csv() function. 1. You can pass in the column name list such as ['name', 'city', 'salary'] to directly read the specified column; 2. You can pass in the column index list such as [0, 2, 3] to read the corresponding column by position, the index starts from 0 and needs to be within the valid range; 3. You can pass in the function such as lambda x: len(x)
If you want to use Pandas to read a specific column from a CSV file, you can use the usecols
parameter of read_csv()
function. This method not only saves memory, but also speeds up reading speed, especially when dealing with large files.

Here is a simple and clear example:
? 1. Basic syntax: read the specified column
Suppose you have a CSV file data.csv
, with the following content:

name,age,city,salary,department Alice, 30, New York, 70,000, Engineering Bob, 25, Los Angeles, 50,000, Sales Charlie, 35, Chicago, 80,000, Engineering Diana, 28, San Francisco, 60,000, Marketing
You only want to read name
, city
and salary
columns:
import pandas as pd df = pd.read_csv('data.csv', usescols=['name', 'city', 'salary']) print(df)
Output result:

name city salary 0 Alice New York 70000 1 Bob Los Angeles 50000 2 Charlie Chicago 80000 3 Diana San Francisco 60000
? 2. Read by column index (suitable for when column names are complicated or if you don’t want to write full names)
You can also read through the column position index , such as reading column 0 (name), column 2 (city), and column 3 (salary):
df = pd.read_csv('data.csv', usecols=[0, 2, 3]) print(df)
The output is the same:
name city salary 0 Alice New York 70000 1 Bob Los Angeles 50000 2 Charlie Chicago 80000 3 Diana San Francisco 60000
?? Note: The index starts at 0, and the index you specify must be within the file column range.
? 3. Use callable to filter column names (advanced usage)
You can also pass in a function that only reads columns that match the rules:
# Read only columns with column name length less than 5 df = pd.read_csv('data.csv', usecols=lambda x: len(x) < 5) print(df)
For example, if only name
and age
meet the conditions (length
? Tips
-
usecols
accepts:- List of column names:
['col1', 'col2']
- Column index list:
[0, 2, 3]
- Function:
lambda x: x in ['name', 'city']
- List of column names:
- If the column name has spaces or case problems, it is recommended to check
pd.read_csv('data.csv', nrows=0).columns
to confirm the column name first. - Using
usecols
can significantly improve performance, especially when skipping dozens of columns and only taking a few columns.
Basically that's it. usecols
is a simple but very practical feature, especially suitable for precise extraction of required fields when handling large CSV files.
The above is the detailed content of python read specific columns from csv pandas example. For more information, please follow other related articles on the PHP Chinese website!

Hot AI Tools

Undress AI Tool
Undress images for free

Undresser.AI Undress
AI-powered app for creating realistic nude photos

AI Clothes Remover
Online AI tool for removing clothes from photos.

Clothoff.io
AI clothes remover

Video Face Swap
Swap faces in any video effortlessly with our completely free AI face swap tool!

Hot Article

Hot Tools

Notepad++7.3.1
Easy-to-use and free code editor

SublimeText3 Chinese version
Chinese version, very easy to use

Zend Studio 13.0.1
Powerful PHP integrated development environment

Dreamweaver CS6
Visual web development tools

SublimeText3 Mac version
God-level code editing software (SublimeText3)

Hot Topics

Polymorphism is a core concept in Python object-oriented programming, referring to "one interface, multiple implementations", allowing for unified processing of different types of objects. 1. Polymorphism is implemented through method rewriting. Subclasses can redefine parent class methods. For example, the spoke() method of Animal class has different implementations in Dog and Cat subclasses. 2. The practical uses of polymorphism include simplifying the code structure and enhancing scalability, such as calling the draw() method uniformly in the graphical drawing program, or handling the common behavior of different characters in game development. 3. Python implementation polymorphism needs to satisfy: the parent class defines a method, and the child class overrides the method, but does not require inheritance of the same parent class. As long as the object implements the same method, this is called the "duck type". 4. Things to note include the maintenance

Parameters are placeholders when defining a function, while arguments are specific values ??passed in when calling. 1. Position parameters need to be passed in order, and incorrect order will lead to errors in the result; 2. Keyword parameters are specified by parameter names, which can change the order and improve readability; 3. Default parameter values ??are assigned when defined to avoid duplicate code, but variable objects should be avoided as default values; 4. args and *kwargs can handle uncertain number of parameters and are suitable for general interfaces or decorators, but should be used with caution to maintain readability.

Iterators are objects that implement __iter__() and __next__() methods. The generator is a simplified version of iterators, which automatically implement these methods through the yield keyword. 1. The iterator returns an element every time he calls next() and throws a StopIteration exception when there are no more elements. 2. The generator uses function definition to generate data on demand, saving memory and supporting infinite sequences. 3. Use iterators when processing existing sets, use a generator when dynamically generating big data or lazy evaluation, such as loading line by line when reading large files. Note: Iterable objects such as lists are not iterators. They need to be recreated after the iterator reaches its end, and the generator can only traverse it once.

A class method is a method defined in Python through the @classmethod decorator. Its first parameter is the class itself (cls), which is used to access or modify the class state. It can be called through a class or instance, which affects the entire class rather than a specific instance; for example, in the Person class, the show_count() method counts the number of objects created; when defining a class method, you need to use the @classmethod decorator and name the first parameter cls, such as the change_var(new_value) method to modify class variables; the class method is different from the instance method (self parameter) and static method (no automatic parameters), and is suitable for factory methods, alternative constructors, and management of class variables. Common uses include:

The key to dealing with API authentication is to understand and use the authentication method correctly. 1. APIKey is the simplest authentication method, usually placed in the request header or URL parameters; 2. BasicAuth uses username and password for Base64 encoding transmission, which is suitable for internal systems; 3. OAuth2 needs to obtain the token first through client_id and client_secret, and then bring the BearerToken in the request header; 4. In order to deal with the token expiration, the token management class can be encapsulated and automatically refreshed the token; in short, selecting the appropriate method according to the document and safely storing the key information is the key.

Python's magicmethods (or dunder methods) are special methods used to define the behavior of objects, which start and end with a double underscore. 1. They enable objects to respond to built-in operations, such as addition, comparison, string representation, etc.; 2. Common use cases include object initialization and representation (__init__, __repr__, __str__), arithmetic operations (__add__, __sub__, __mul__) and comparison operations (__eq__, ___lt__); 3. When using it, make sure that their behavior meets expectations. For example, __repr__ should return expressions of refactorable objects, and arithmetic methods should return new instances; 4. Overuse or confusing things should be avoided.

Pythonmanagesmemoryautomaticallyusingreferencecountingandagarbagecollector.Referencecountingtrackshowmanyvariablesrefertoanobject,andwhenthecountreacheszero,thememoryisfreed.However,itcannothandlecircularreferences,wheretwoobjectsrefertoeachotherbuta

Python's garbage collection mechanism automatically manages memory through reference counting and periodic garbage collection. Its core method is reference counting, which immediately releases memory when the number of references of an object is zero; but it cannot handle circular references, so a garbage collection module (gc) is introduced to detect and clean the loop. Garbage collection is usually triggered when the reference count decreases during program operation, the allocation and release difference exceeds the threshold, or when gc.collect() is called manually. Users can turn off automatic recycling through gc.disable(), manually execute gc.collect(), and adjust thresholds to achieve control through gc.set_threshold(). Not all objects participate in loop recycling. If objects that do not contain references are processed by reference counting, it is built-in
