擅长:python、mysql、java
<p>如果知道前面的列名,可以通过设置<code>usecols</code>参数来实现</p>
<p><strong>当您知道要使用哪些列时</p>
<p>假设您有一个csv文件,其中有<code>['id','name','last_name']</code>列,您只需要<code>['name','last_name']</code>。您可以按以下步骤操作:</p>
<pre><code>import pandas as pd
df = pd.read_csv("sample.csv", usecols = ['name','last_name'])
</code></pre>
<p><strong>需要前N列时</strong></p>
<p>如果您不知道列名,但需要dataframe中的前N列。你可以通过</p>
<pre><code>import pandas as pd
df = pd.read_csv("sample.csv", usecols = [i for i in range(n)])
</code></pre>
<p><strong>编辑</strong></p>
<p><strong>当您知道要删除的列的名称时</strong></p>
<pre><code># Read column names from file
cols = list(pd.read_csv("sample_data.csv", nrows =1))
print(cols)
# Use list comprehension to remove the unwanted column in **usecol**
df= pd.read_csv("sample_data.csv", usecols =[i for i in cols if i != 'name'])
</code></pre>