如何从csv文件中选择3列系数的数据

Question

我想根据我的数据集中的行索引，最好通过Pandas.DataFrame 绘制2个不同场景的列数：

第一种情况： 列 index[2,5,8,..., n+2]

第二种情况：最后 480 列或列索引 [961-1439] picture

我尝试使用以下列的索引：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

dft = pd.read_csv("D:\Test.csv" , header=None)
dft.head()

id_set = dft[dft.index % 2 == 0].astype('int').values
A = dft[dft.index % 2 == 1].values
B = dft[dft.index % 2 == 2].values
C = dft[dft.index % 2 == 3].values
data = {'A': A[:,0], 'B': B[:,0], 'C': C[:,0]}
df = pd.DataFrame(data, columns=['A','B','C'], index = id_set[:,0])

#1st scenario
j=0
index=[]
for i in range(1439):
    if j==2:
        j=0
        continue
    else:
        index.append(i)
        j+=1
print(index)

#2nd scenario
last_480 = df.[0:480][::-1]

我找到了 post1 和 post2，但它们不是我的情况！

如果有人可以帮助我，我将不胜感激。

Answer 1

第一种情况：

df.iloc[:, 2::3]

这里的切片是指从第2个开始的所有行、列，之后的每3个。

第二种情况：

df.iloc[:, :961:-1]

这里的切片是指从列表末尾到961的所有行、列。

编辑：

import matplotlib.pyplot as plt
import seaborn as sns

senario1 = df.iloc[:, 2::3].copy()
sns.lineplot(data = senario1.T)

您可以将切片的副本保存到另一个变量，然后由于您想按行绘制图形，您需要对切片矩阵进行转置（这将使您的行变成列）。